Spaces:
Paused
Paused
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 |
|