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

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-08 17:22:44 +08:00
parent 827a0b2c55
commit 4ef813555c
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-debug.py
View file

@ -121,7 +121,7 @@ def main():
# latent_size = vae.get_latent_size(input_size)
# 3.2. move to device & eval
vae = vae.to(device, dtype)
vae = vae.to(device, dtype).eval()
# # 4.5. setup optimizer
# optimizer = HybridAdam(