RamAnanth1 commited on
Commit
1b1d216
β€’
1 Parent(s): b1693d8

Update app.py

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