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

update config

Browse source
This commit is contained in:
zhengzangw 2024-05-05 16:06:30 +00:00
parent f77370fff6
commit f0c4d1bebd
1 changed files with 3 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
configs/opensora-v1-2/train/adapt-vae.py
View file

@ -7,7 +7,7 @@ bucket_config = {
"144p": {1: (1.0, 64)},
# ---
"256": {1: (0.5, 48)},
"240p": {1: (0.5, 48)},
"240p": {1: (0.5, 48), 16: (1.0, 8)},
# ---
"360p": {1: (0.5, 18)},
"512": {1: (0.5, 18)},
@ -38,16 +38,10 @@ model = dict(
enable_flashattn=True,
enable_layernorm_kernel=True,
)
# vae = dict(
# type="VideoAutoencoderKL",
# micro_batch_size=4,
# from_pretrained="PixArt-alpha/pixart_sigma_sdxlvae_T5_diffusers",
# subfolder="vae",
# local_files_only=True,
# )
vae = dict(
type="VideoAutoencoderPipeline",
from_pretrained="pretrained_models/vae-v1",
from_pretrained="pretrained_models/vae-v2",
micro_frame_size=16,
vae_2d=dict(
type="VideoAutoencoderKL",
from_pretrained="PixArt-alpha/pixart_sigma_sdxlvae_T5_diffusers",