nekoshadow commited on
Commit
7822350
1 Parent(s): 6ad2dc4

Fix import issue

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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