Anton Lozhkov
|
761f0297b0
|
[Tests] Fix spatial transformer tests on GPU (#531)
|
2022-09-16 16:04:37 +02:00 |
|
Sid Sahai
|
f73ca908e5
|
[Tests] Test attention.py (#368)
* add test for AttentionBlock, SpatialTransformer
* add context_dim, handle device
* removed dropout test
* fixes, add dropout test
|
2022-09-16 12:59:42 +02:00 |
|
Anton Lozhkov
|
ed22b4fd07
|
Revive `make quality` (#203)
* Revive Make utils
* Add datasets for training too
|
2022-08-17 15:22:04 +02:00 |
|
Suraj Patil
|
4e2674934f
|
add tests for 1D Up/Downsample blocks (#72)
|
2022-07-04 11:41:04 +02:00 |
|
Patrick von Platen
|
321f9791d6
|
Downsample / Upsample - clean to 1D and 2D (#68)
* make unet rl work
* uploaad files / code
* upload files
* make style correct
* finish
|
2022-07-03 22:26:33 +02:00 |
|
patil-suraj
|
7b9b946cb2
|
add tests for downsample block
|
2022-06-27 18:03:51 +02:00 |
|
patil-suraj
|
dc7c49e4e4
|
add tests for upsample blocks
|
2022-06-27 15:50:54 +02:00 |
|
Patrick von Platen
|
c7a39d38ad
|
refactor all sinus embeddings
|
2022-06-27 11:37:37 +00:00 |
|
Patrick von Platen
|
02a76c2c81
|
consolidate timestep embeds
|
2022-06-27 10:14:54 +00:00 |
|
Patrick von Platen
|
45a09bebf3
|
add first files
|
2022-06-27 10:46:39 +02:00 |