Spaces:
Runtime error
Runtime error
use gpu
Browse files
config/ldm_autoencoder_kl.yaml
CHANGED
@@ -28,5 +28,5 @@ model:
|
|
28 |
lightning:
|
29 |
trainer:
|
30 |
benchmark: True
|
31 |
-
|
32 |
devices: 1
|
|
|
28 |
lightning:
|
29 |
trainer:
|
30 |
benchmark: True
|
31 |
+
accelerator: gpu
|
32 |
devices: 1
|