Hugging Face's logo

Spaces:
Vchitect
/
LaVie
Running on A10G

App Files Community
4
Zhouyan248 commited on
Commit
3066241
•
1 Parent(s): 88e4948

Update base/text_to_video/__init__.py

Browse files
Files changed (1) hide show
  1. base/text_to_video/__init__.py +1 -1
base/text_to_video/__init__.py CHANGED
@@ -25,7 +25,7 @@ def model_t2v_fun(args):
25
  # sd_path = args.pretrained_path + "/stable-diffusion-v1-4"
26
  sd_path = args.pretrained_path
27
  unet = get_models(args, sd_path).to(device, dtype=torch.float16)
28
- state_dict = find_model(args.pretrained_path + "/lavie_base.pt")
29
  # state_dict = find_model("./pretrained_models/lavie_base.pt")
30
  unet.load_state_dict(state_dict)
31
 
 
25
  # sd_path = args.pretrained_path + "/stable-diffusion-v1-4"
26
  sd_path = args.pretrained_path
27
  unet = get_models(args, sd_path).to(device, dtype=torch.float16)
28
+ state_dict = find_model("../pretrained_models/lavie_base.pt")
29
  # state_dict = find_model("./pretrained_models/lavie_base.pt")
30
  unet.load_state_dict(state_dict)
31