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

inference v2 working

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-16 18:30:31 +08:00
parent 179bb7b125
commit 95517d7fb5
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/inference-vae-v2.py
View file

@ -70,7 +70,7 @@ def main():
dataset,
batch_size=cfg.batch_size,
num_workers=cfg.num_workers,
shuffle=True,
shuffle=False,
drop_last=True,
pin_memory=True,
process_group=get_data_parallel_group(),
@ -187,7 +187,7 @@ def main():
adversarial_loss = adversarial_loss_fn(
fake_logits,
nll_loss,
vae.module.get_last_layer(),
vae.get_last_layer(),
cfg.discriminator_start+1, # Hack to use discriminator
is_training = vae.training,
)