Spaces:
Runtime error
Runtime error
Commit
·
758cfc5
1
Parent(s):
928a499
Fix EOF literal
Browse files
model.py
CHANGED
|
@@ -63,7 +63,7 @@ class Model:
|
|
| 63 |
self.model_dir.mkdir(exist_ok=True, parents=True)
|
| 64 |
device = 'cuda'
|
| 65 |
|
| 66 |
-
config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml)
|
| 67 |
config.model.params.cond_stage_config.params.device = device
|
| 68 |
|
| 69 |
base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
|
|
|
|
| 63 |
self.model_dir.mkdir(exist_ok=True, parents=True)
|
| 64 |
device = 'cuda'
|
| 65 |
|
| 66 |
+
config = OmegaConf.load("configs/stable-diffusion/test_sketch.yaml")
|
| 67 |
config.model.params.cond_stage_config.params.device = device
|
| 68 |
|
| 69 |
base_model_file = "https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt"
|