Chao Xu commited on
Commit
7d7aedf
โ€ข
1 Parent(s): 09ac26d

update ckpt path

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -491,7 +491,7 @@ def run_demo(
491
  print('new device_idx:', device_idx)
492
 
493
  device = f"cuda:{device_idx}" if torch.cuda.is_available() else "cpu"
494
- models = init_model(device, ckpt)
495
  # model = models['turncam']
496
  # sampler = DDIMSampler(model)
497
 
 
491
  print('new device_idx:', device_idx)
492
 
493
  device = f"cuda:{device_idx}" if torch.cuda.is_available() else "cpu"
494
+ models = init_model(device, os.path.join(code_dir, ckpt))
495
  # model = models['turncam']
496
  # sampler = DDIMSampler(model)
497