nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-12 13:54:53 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

updated config

Browse source
This commit is contained in:
zhengzangw 2024-05-02 05:41:39 +00:00
parent 02904b4f29
commit 4f915a51dd
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
configs/vae/train/video.py
View file

@ -19,7 +19,7 @@ plugin = "zero2"
# Define model
model = dict(
type="VideoAutoencoderPipeline",
freeze_vae_2d=True,
freeze_vae_2d=False,
from_pretrained=None,
vae_2d=dict(
type="VideoAutoencoderKL",
@ -39,7 +39,7 @@ kl_loss_weight = 1e-6
mixed_image_ratio = 0.2
use_real_rec_loss = True
use_z_rec_loss = True
use_z_rec_loss = False
use_image_identity_loss = False
# Others