tests/test_training.py · 64270eff34381fd02b21cf3129e38b0da155c81a · OpenDAS / diffusers · GitLab

"...composable_kernel_rocm.git" did not exist on "be3fbf7f5f3bdc3e1668490fb93b8a44aa2638ce"

Find file Blame History Permalink

Add training and batched inference test for DDPM vs DDIM (#140) · 6c15636b
Anton Lozhkov authored Jul 27, 2022
```
* Add torch_device to the VE pipeline

* Mark the training test with slow
```
6c15636b

test_training.py 3.37 KB