Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
8c4bc8e
1
Parent(s):
448d195
Update base/app.py
Browse files- base/app.py +1 -1
base/app.py
CHANGED
@@ -31,7 +31,7 @@ h1 {
|
|
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)
|
|
|
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)
|