Hugging Face's logo

Spaces:
Vchitect
/
LaVie
Running on A10G

App Files Community
4
Zhouyan248 commited on
Commit
8c4bc8e
•
1 Parent(s): 448d195

Update base/app.py

Browse files
Files changed (1) hide show
  1. base/app.py +1 -1
base/app.py CHANGED
@@ -31,7 +31,7 @@ h1 {
31
  }
32
  """
33
 
34
- sd_path = args.pretrained_path + "/stable-diffusion-v1-4"
35
  unet = get_models(args, sd_path).to(device, dtype=torch.float16)
36
  state_dict = find_model("./pretrained_models/lavie_base.pt")
37
  unet.load_state_dict(state_dict)
 
31
  }
32
  """
33
 
34
+ sd_path = args.pretrained_path
35
  unet = get_models(args, sd_path).to(device, dtype=torch.float16)
36
  state_dict = find_model("./pretrained_models/lavie_base.pt")
37
  unet.load_state_dict(state_dict)