Logic/DDT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bugs(update configs)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
configs/repa_improved_ddt_xlen22de6_256.yaml
View File

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

6
configs/repa_improved_ddt_xlen22de6_512.yaml
View File

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

6
configs/repa_improved_dit_large.yaml
View File

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

6
configs/repa_improved_dit_xl.yaml
View File

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