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

decoupled dit configs

Browse Source
This commit is contained in:
wangshuai6
2025-04-09 11:14:04 +08:00
parent d1b6da1f0a
commit 4f627dbc75
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
configs/repa_flatten_condit22_fixt_xl.yaml → configs/repa_improved_ddt_xlen22de6_256.yaml
View File

@@ -42,14 +42,14 @@ model:
precompute: true precompute: true
weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/ weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
denoiser: denoiser:
class_path: src.models.denoiser.flatten_condit_dit_fixt.FlattenConDiT class_path: src.models.denoiser.decoupled_improved_dit.DDT
init_args: init_args:
in_channels: 4 in_channels: 4
patch_size: 2 patch_size: 2
num_groups: 16 num_groups: 16
hidden_size: &hidden_dim 1152 hidden_size: &hidden_dim 1152
num_blocks: 28 num_blocks: 28
num_cond_blocks: 22 num_encoder_blocks: 22
num_classes: 1000 num_classes: 1000
conditioner: conditioner:
class_path: src.models.conditioner.LabelConditioner class_path: src.models.conditioner.LabelConditioner

4
configs/repa_flatten_condit22_fixt_xl512.yaml → configs/repa_improved_ddt_xlen22de6_512.yaml
View File

@@ -42,14 +42,14 @@ model:
precompute: true precompute: true
weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/ weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
denoiser: denoiser:
class_path: src.models.denoiser.flatten_condit_dit_fixt.FlattenConDiT class_path: src.models.denoiser.decoupled_improved_dit.DDT
init_args: init_args:
in_channels: 4 in_channels: 4
patch_size: 2 patch_size: 2
num_groups: 16 num_groups: 16
hidden_size: &hidden_dim 1152 hidden_size: &hidden_dim 1152
num_blocks: 28 num_blocks: 28
num_cond_blocks: 22 num_encoder_blocks: 22
num_classes: 1000 num_classes: 1000
conditioner: conditioner:
class_path: src.models.conditioner.LabelConditioner class_path: src.models.conditioner.LabelConditioner

4
configs/repa_flatten_dit_fixt_large.yaml → configs/repa_improved_dit_large.yaml
View File

@@ -1,7 +1,7 @@
# lightning.pytorch==2.4.0 # lightning.pytorch==2.4.0
seed_everything: true seed_everything: true
tags: tags:
exp: &exp repa_flatten_dit_fixt_large exp: &exp repa_improved_dit_large
torch_hub_dir: /mnt/bn/wangshuai6/torch_hub torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
huggingface_cache_dir: null huggingface_cache_dir: null
trainer: trainer:
@@ -42,7 +42,7 @@ model:
precompute: true precompute: true
weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/ weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
denoiser: denoiser:
class_path: src.models.denoiser.flatten_dit_fixt.FlattenDiT class_path: src.models.denoiser.improved_dit.DiT
init_args: init_args:
in_channels: 4 in_channels: 4
patch_size: 2 patch_size: 2

4
configs/repa_flatten_dit_fixt_xl.yaml → configs/repa_improved_dit_xl.yaml
View File

@@ -1,7 +1,7 @@
# lightning.pytorch==2.4.0 # lightning.pytorch==2.4.0
seed_everything: true seed_everything: true
tags: tags:
exp: &exp repa_flatten_dit_fixt_xl exp: &exp repa_improved_dit_xlen22de6_512
torch_hub_dir: /mnt/bn/wangshuai6/torch_hub torch_hub_dir: /mnt/bn/wangshuai6/torch_hub
huggingface_cache_dir: null huggingface_cache_dir: null
trainer: trainer:
@@ -42,7 +42,7 @@ model:
precompute: true precompute: true
weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/ weight_path: /mnt/bn/wangshuai6/models/sd-vae-ft-ema/
denoiser: denoiser:
class_path: src.models.denoiser.flatten_dit_fixt.FlattenDiT class_path: src.models.denoiser.improved_dit.DiT
init_args: init_args:
in_channels: 4 in_channels: 4
patch_size: 2 patch_size: 2