fix bugs(update configs)

This commit is contained in:
wangshuai6
2025-04-13 09:23:09 +08:00
parent ce93cc61a8
commit a307da32e1
4 changed files with 12 additions and 12 deletions

View File

@@ -2,10 +2,10 @@
seed_everything: true
tags:
exp: &exp repa_improved_dit_xlen22de6_512
torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
torch_hub_dir: null
huggingface_cache_dir: null
trainer:
default_root_dir: /mnt/bn/wangshuai6/universal_flow_workdirs
default_root_dir: universal_flow_workdirs
accelerator: auto
strategy: auto
devices: auto
@@ -40,7 +40,7 @@ model:
class_path: src.models.vae.LatentVAE
init_args:
precompute: true
weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
weight_path: stabilityai/sd-vae-ft-ema
denoiser:
class_path: src.models.denoiser.improved_dit.DiT
init_args: