Spaces:
Runtime error
Runtime error
Chao Xu
commited on
Commit
โข
7d7aedf
1
Parent(s):
09ac26d
update ckpt path
Browse files
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 |
|