nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-14 16:45:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-08 16:14:00 +08:00
parent 11764f2b42
commit 90fc92ee9c
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/inference-vae.py
View file

@ -145,7 +145,7 @@ def main():
model=vae, dataloader=dataloader
)
# load model using booster
booster.load_model(vae, os.path.join(cfg.ckpt_path, "model"))
booster.load_model(vae, os.path.join(cfg.model["from_pretrained"], "model"))
# 4.1. batch generation