Spaces:
Paused
Paused
nekoshadow
commited on
Commit
•
7822350
1
Parent(s):
6ad2dc4
Fix import issue
Browse files
app.py
CHANGED
@@ -171,8 +171,8 @@ def run_demo():
|
|
171 |
|
172 |
if deployed:
|
173 |
controlnet = ControlNetModelSync.from_pretrained('controlnet_ckpt', torch_dtype=torch.float32, use_safetensors=True)
|
174 |
-
cfg = 'configs/syncdreamer.yaml'
|
175 |
-
dreamer = load_model(cfg, 'ckpt/syncdreamer-pretrain.ckpt', strict=True)
|
176 |
|
177 |
controlnet.to('cuda', dtype=torch.float32)
|
178 |
|
|
|
171 |
|
172 |
if deployed:
|
173 |
controlnet = ControlNetModelSync.from_pretrained('controlnet_ckpt', torch_dtype=torch.float32, use_safetensors=True)
|
174 |
+
cfg = 'SyncDreamer/configs/syncdreamer.yaml'
|
175 |
+
dreamer = load_model(cfg, 'SyncDreamer/ckpt/syncdreamer-pretrain.ckpt', strict=True)
|
176 |
|
177 |
controlnet.to('cuda', dtype=torch.float32)
|
178 |
|