Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
β’
1b1d216
1
Parent(s):
b1693d8
Update app.py
Browse files
app.py
CHANGED
@@ -49,6 +49,7 @@ model, _, _ = load_model(config, ckpt_path,
|
|
49 |
inject_lora=False,
|
50 |
lora_scale=None,
|
51 |
)
|
|
|
52 |
state_dict = torch.load(adapter_ckpt, map_location="cpu")
|
53 |
if "state_dict" in list(state_dict.keys()):
|
54 |
state_dict = state_dict["state_dict"]
|
|
|
49 |
inject_lora=False,
|
50 |
lora_scale=None,
|
51 |
)
|
52 |
+
adapter_ckpt = 'models/adapter_t2v_depth/adapter.pth'
|
53 |
state_dict = torch.load(adapter_ckpt, map_location="cpu")
|
54 |
if "state_dict" in list(state_dict.keys()):
|
55 |
state_dict = state_dict["state_dict"]
|