Upload folder using huggingface_hub
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +56 -0
- dlay_flux2_loha_1gpu_v2/.ipynb_checkpoints/config-checkpoint.yaml +105 -0
- dlay_flux2_loha_1gpu_v2/config.yaml +105 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000000500.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000001000.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000001500.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000002000.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000002500.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000003000.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000003500.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000004000.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000004500.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000005000.safetensors +3 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963831954__000000500_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963885253__000000500_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963937896__000000500_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966287976__000001000_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966340638__000001000_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966393280__000001000_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968750269__000001500_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968802942__000001500_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968855619__000001500_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971191622__000002000_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971244311__000002000_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971297007__000002000_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973637152__000002500_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973689848__000002500_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973742557__000002500_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976089753__000003000_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976142455__000003000_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976195153__000003000_2-checkpoint.jpg +3 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978551151__000003500_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978603817__000003500_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978656496__000003500_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764980981509__000004000_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764981034171__000004000_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764981086832__000004000_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983432333__000004500_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983484988__000004500_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983537656__000004500_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985882360__000005000_0-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985935041__000005000_1-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985987725__000005000_2-checkpoint.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764963831954__000000500_0.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764963885253__000000500_1.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764963937896__000000500_2.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764966287976__000001000_0.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764966340638__000001000_1.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764966393280__000001000_2.jpg +0 -0
- dlay_flux2_loha_1gpu_v2/samples/1764968750269__000001500_0.jpg +0 -0
.gitattributes
CHANGED
|
@@ -89,3 +89,59 @@ dlay_flux2_lora_fast_v1/samples/1764726440101__000004000_3.jpg filter=lfs diff=l
|
|
| 89 |
dlay_flux2_lora_fast_v1/samples/1764726451078__000004000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 90 |
dlay_flux2_lora_fast_v1/samples/1764726461979__000004000_5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 91 |
dlay_flux2_lora_fast_v1/samples/1764726472987__000004000_6.jpg filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 89 |
dlay_flux2_lora_fast_v1/samples/1764726451078__000004000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 90 |
dlay_flux2_lora_fast_v1/samples/1764726461979__000004000_5.jpg filter=lfs diff=lfs merge=lfs -text
|
| 91 |
dlay_flux2_lora_fast_v1/samples/1764726472987__000004000_6.jpg filter=lfs diff=lfs merge=lfs -text
|
| 92 |
+
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976195153__000003000_2-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 93 |
+
dlay_flux2_loha_1gpu_v2/samples/1764976195153__000003000_2.jpg filter=lfs diff=lfs merge=lfs -text
|
| 94 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764783621256__000000500_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 95 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764783644033__000000500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 96 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764784965936__000001000_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 97 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764784988736__000001000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 98 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764786310046__000001500_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 99 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764786332854__000001500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 100 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764790015322__000002500_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 101 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764792775816__000003500_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 102 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/.ipynb_checkpoints/1764792799808__000003500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 103 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764783621256__000000500_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 104 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764783644033__000000500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 105 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764784965936__000001000_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 106 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764784988736__000001000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 107 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764786310046__000001500_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 108 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764786332854__000001500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 109 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764788635681__000002000_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 110 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764788659719__000002000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 111 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764790015322__000002500_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 112 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764790039320__000002500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 113 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764791380438__000003000_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 114 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764791404439__000003000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 115 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764792775816__000003500_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 116 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/1764792799808__000003500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 117 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/dlay100.png filter=lfs diff=lfs merge=lfs -text
|
| 118 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/dlay118.png filter=lfs diff=lfs merge=lfs -text
|
| 119 |
+
dlay_flux2_lokr_optimized_v1-adamw-offload/samples/dlay260.png filter=lfs diff=lfs merge=lfs -text
|
| 120 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764852970108__000001000_3-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 121 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764852994558__000001000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 122 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764857209719__000002500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 123 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764858610090__000003000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 124 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764860020632__000003500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 125 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764860069478__000003500_6-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 126 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764861432243__000004000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 127 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764866408110__000005000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 128 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764867874291__000005500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 129 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764869336235__000006000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 130 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764873721495__000007500_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 131 |
+
dlay_flux2_lokr_subject_v2/samples/.ipynb_checkpoints/1764875169171__000008000_4-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
|
| 132 |
+
dlay_flux2_lokr_subject_v2/samples/1764852970108__000001000_3.jpg filter=lfs diff=lfs merge=lfs -text
|
| 133 |
+
dlay_flux2_lokr_subject_v2/samples/1764852994558__000001000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 134 |
+
dlay_flux2_lokr_subject_v2/samples/1764854406631__000001500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 135 |
+
dlay_flux2_lokr_subject_v2/samples/1764855815671__000002000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 136 |
+
dlay_flux2_lokr_subject_v2/samples/1764857209719__000002500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 137 |
+
dlay_flux2_lokr_subject_v2/samples/1764858610090__000003000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 138 |
+
dlay_flux2_lokr_subject_v2/samples/1764860020632__000003500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 139 |
+
dlay_flux2_lokr_subject_v2/samples/1764860069478__000003500_6.jpg filter=lfs diff=lfs merge=lfs -text
|
| 140 |
+
dlay_flux2_lokr_subject_v2/samples/1764861432243__000004000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 141 |
+
dlay_flux2_lokr_subject_v2/samples/1764866408110__000005000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 142 |
+
dlay_flux2_lokr_subject_v2/samples/1764867874291__000005500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 143 |
+
dlay_flux2_lokr_subject_v2/samples/1764869336235__000006000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 144 |
+
dlay_flux2_lokr_subject_v2/samples/1764870790803__000006500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 145 |
+
dlay_flux2_lokr_subject_v2/samples/1764872262983__000007000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 146 |
+
dlay_flux2_lokr_subject_v2/samples/1764873721495__000007500_4.jpg filter=lfs diff=lfs merge=lfs -text
|
| 147 |
+
dlay_flux2_lokr_subject_v2/samples/1764875169171__000008000_4.jpg filter=lfs diff=lfs merge=lfs -text
|
dlay_flux2_loha_1gpu_v2/.ipynb_checkpoints/config-checkpoint.yaml
ADDED
|
@@ -0,0 +1,105 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
job: extension
|
| 2 |
+
config:
|
| 3 |
+
name: dlay_flux2_loha_1gpu_v2
|
| 4 |
+
process:
|
| 5 |
+
- type: sd_trainer
|
| 6 |
+
training_folder: /root/output
|
| 7 |
+
performance_log_every: 100
|
| 8 |
+
device: cuda:0
|
| 9 |
+
network:
|
| 10 |
+
type: loha
|
| 11 |
+
linear: 128
|
| 12 |
+
linear_alpha: 64
|
| 13 |
+
dropout: 0.0
|
| 14 |
+
use_tucker: false
|
| 15 |
+
use_scalar: false
|
| 16 |
+
weight_decompose: false
|
| 17 |
+
wd_on_out: false
|
| 18 |
+
rs_lora: false
|
| 19 |
+
rank_dropout_scale: false
|
| 20 |
+
network_kwargs:
|
| 21 |
+
rank_dropout: 0.02
|
| 22 |
+
model:
|
| 23 |
+
name_or_path: black-forest-labs/FLUX.2-dev
|
| 24 |
+
arch: flux2
|
| 25 |
+
quantize: false
|
| 26 |
+
quantize_te: false
|
| 27 |
+
low_vram: false
|
| 28 |
+
layer_offloading: true
|
| 29 |
+
layer_offloading_transformer_percent: 0.75
|
| 30 |
+
datasets:
|
| 31 |
+
- folder_path: /root/DATASETS/DLAY/flux2
|
| 32 |
+
caption_ext: txt
|
| 33 |
+
default_caption: DLAY man
|
| 34 |
+
mask_path: /root/DATASETS/DLAY/mask
|
| 35 |
+
mask_min_value: 0.0
|
| 36 |
+
invert_mask: false
|
| 37 |
+
loss_multiplier: 1.2
|
| 38 |
+
resolution:
|
| 39 |
+
- 512
|
| 40 |
+
- 768
|
| 41 |
+
- 1024
|
| 42 |
+
cache_latents: true
|
| 43 |
+
cache_latents_to_disk: true
|
| 44 |
+
cache_text_embeddings: true
|
| 45 |
+
caption_dropout_rate: 0.05
|
| 46 |
+
shuffle_tokens: true
|
| 47 |
+
train:
|
| 48 |
+
batch_size: 2
|
| 49 |
+
gradient_accumulation: 1
|
| 50 |
+
steps: 10000
|
| 51 |
+
optimizer: prodigy
|
| 52 |
+
lr: 1.0
|
| 53 |
+
optimizer_params:
|
| 54 |
+
d_coef: 1.5
|
| 55 |
+
use_bias_correction: true
|
| 56 |
+
weight_decay: 0.01
|
| 57 |
+
safeguard_warmup: true
|
| 58 |
+
betas:
|
| 59 |
+
- 0.9
|
| 60 |
+
- 0.999
|
| 61 |
+
lr_scheduler: cosine_with_warmup
|
| 62 |
+
lr_scheduler_params:
|
| 63 |
+
num_warmup_steps: 300
|
| 64 |
+
num_training_steps: 6000
|
| 65 |
+
dtype: bf16
|
| 66 |
+
attention_backend: flash
|
| 67 |
+
gradient_checkpointing: false
|
| 68 |
+
noise_scheduler: flowmatch
|
| 69 |
+
timestep_type: weighted
|
| 70 |
+
max_grad_norm: 1.0
|
| 71 |
+
min_snr_gamma: null
|
| 72 |
+
ema_config:
|
| 73 |
+
use_ema: false
|
| 74 |
+
train_unet: true
|
| 75 |
+
train_text_encoder: false
|
| 76 |
+
skip_first_sample: true
|
| 77 |
+
cache_text_embeddings: true
|
| 78 |
+
unload_text_encoder: true
|
| 79 |
+
diff_output_preservation: false
|
| 80 |
+
diff_output_preservation_multiplier: 1.0
|
| 81 |
+
diff_output_preservation_class: man
|
| 82 |
+
save:
|
| 83 |
+
dtype: bfloat16
|
| 84 |
+
save_every: 500
|
| 85 |
+
save_format: safetensors
|
| 86 |
+
max_step_saves_to_keep: 10
|
| 87 |
+
sample:
|
| 88 |
+
sampler: flowmatch
|
| 89 |
+
sample_every: 500
|
| 90 |
+
width: 1024
|
| 91 |
+
height: 1024
|
| 92 |
+
prompts:
|
| 93 |
+
- DLAY man, portrait, neutral background, studio lighting
|
| 94 |
+
- DLAY man, corporate headshot, navy suit, professional lighting
|
| 95 |
+
- DLAY man, holding coffee, cafe interior, warm ambient lighting
|
| 96 |
+
neg: ''
|
| 97 |
+
seed: 42
|
| 98 |
+
walk_seed: true
|
| 99 |
+
guidance_scale: 4.0
|
| 100 |
+
sample_steps: 28
|
| 101 |
+
logging:
|
| 102 |
+
log_every: 10
|
| 103 |
+
use_wandb: true
|
| 104 |
+
wandb_project: flux2-dlay-b300
|
| 105 |
+
wandb_run_name: dlay-flux2-loha-1gpu-v2
|
dlay_flux2_loha_1gpu_v2/config.yaml
ADDED
|
@@ -0,0 +1,105 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
job: extension
|
| 2 |
+
config:
|
| 3 |
+
name: dlay_flux2_loha_1gpu_v2
|
| 4 |
+
process:
|
| 5 |
+
- type: sd_trainer
|
| 6 |
+
training_folder: /root/output
|
| 7 |
+
performance_log_every: 100
|
| 8 |
+
device: cuda:0
|
| 9 |
+
network:
|
| 10 |
+
type: loha
|
| 11 |
+
linear: 128
|
| 12 |
+
linear_alpha: 64
|
| 13 |
+
dropout: 0.0
|
| 14 |
+
use_tucker: false
|
| 15 |
+
use_scalar: false
|
| 16 |
+
weight_decompose: false
|
| 17 |
+
wd_on_out: false
|
| 18 |
+
rs_lora: false
|
| 19 |
+
rank_dropout_scale: false
|
| 20 |
+
network_kwargs:
|
| 21 |
+
rank_dropout: 0.02
|
| 22 |
+
model:
|
| 23 |
+
name_or_path: black-forest-labs/FLUX.2-dev
|
| 24 |
+
arch: flux2
|
| 25 |
+
quantize: false
|
| 26 |
+
quantize_te: false
|
| 27 |
+
low_vram: false
|
| 28 |
+
layer_offloading: true
|
| 29 |
+
layer_offloading_transformer_percent: 0.75
|
| 30 |
+
datasets:
|
| 31 |
+
- folder_path: /root/DATASETS/DLAY/flux2
|
| 32 |
+
caption_ext: txt
|
| 33 |
+
default_caption: DLAY man
|
| 34 |
+
mask_path: /root/DATASETS/DLAY/mask
|
| 35 |
+
mask_min_value: 0.0
|
| 36 |
+
invert_mask: false
|
| 37 |
+
loss_multiplier: 1.2
|
| 38 |
+
resolution:
|
| 39 |
+
- 512
|
| 40 |
+
- 768
|
| 41 |
+
- 1024
|
| 42 |
+
cache_latents: true
|
| 43 |
+
cache_latents_to_disk: true
|
| 44 |
+
cache_text_embeddings: true
|
| 45 |
+
caption_dropout_rate: 0.05
|
| 46 |
+
shuffle_tokens: true
|
| 47 |
+
train:
|
| 48 |
+
batch_size: 2
|
| 49 |
+
gradient_accumulation: 1
|
| 50 |
+
steps: 10000
|
| 51 |
+
optimizer: prodigy
|
| 52 |
+
lr: 1.0
|
| 53 |
+
optimizer_params:
|
| 54 |
+
d_coef: 1.5
|
| 55 |
+
use_bias_correction: true
|
| 56 |
+
weight_decay: 0.01
|
| 57 |
+
safeguard_warmup: true
|
| 58 |
+
betas:
|
| 59 |
+
- 0.9
|
| 60 |
+
- 0.999
|
| 61 |
+
lr_scheduler: cosine_with_warmup
|
| 62 |
+
lr_scheduler_params:
|
| 63 |
+
num_warmup_steps: 300
|
| 64 |
+
num_training_steps: 6000
|
| 65 |
+
dtype: bf16
|
| 66 |
+
attention_backend: flash
|
| 67 |
+
gradient_checkpointing: false
|
| 68 |
+
noise_scheduler: flowmatch
|
| 69 |
+
timestep_type: weighted
|
| 70 |
+
max_grad_norm: 1.0
|
| 71 |
+
min_snr_gamma: null
|
| 72 |
+
ema_config:
|
| 73 |
+
use_ema: false
|
| 74 |
+
train_unet: true
|
| 75 |
+
train_text_encoder: false
|
| 76 |
+
skip_first_sample: true
|
| 77 |
+
cache_text_embeddings: true
|
| 78 |
+
unload_text_encoder: true
|
| 79 |
+
diff_output_preservation: false
|
| 80 |
+
diff_output_preservation_multiplier: 1.0
|
| 81 |
+
diff_output_preservation_class: man
|
| 82 |
+
save:
|
| 83 |
+
dtype: bfloat16
|
| 84 |
+
save_every: 500
|
| 85 |
+
save_format: safetensors
|
| 86 |
+
max_step_saves_to_keep: 10
|
| 87 |
+
sample:
|
| 88 |
+
sampler: flowmatch
|
| 89 |
+
sample_every: 500
|
| 90 |
+
width: 1024
|
| 91 |
+
height: 1024
|
| 92 |
+
prompts:
|
| 93 |
+
- DLAY man, portrait, neutral background, studio lighting
|
| 94 |
+
- DLAY man, corporate headshot, navy suit, professional lighting
|
| 95 |
+
- DLAY man, holding coffee, cafe interior, warm ambient lighting
|
| 96 |
+
neg: ''
|
| 97 |
+
seed: 42
|
| 98 |
+
walk_seed: true
|
| 99 |
+
guidance_scale: 4.0
|
| 100 |
+
sample_steps: 28
|
| 101 |
+
logging:
|
| 102 |
+
log_every: 10
|
| 103 |
+
use_wandb: true
|
| 104 |
+
wandb_project: flux2-dlay-b300
|
| 105 |
+
wandb_run_name: dlay-flux2-loha-1gpu-v2
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000000500.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:19582c75e48bfe74c643c3db79e631591213889ced501f9ab7378b601b2ba5fa
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000001000.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:24a7b14567978f84f2f26a904dd6d20a5ebd6fdb97808bf7f9b5442436ffaae1
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000001500.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9dd017b648cce76c0d2f5f0ebe11218a175d482e39e73bd5ea96b0c4d550dbe6
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000002000.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2abf87a40837556d31f62d4775ab42a09ec25d0ca3d6948d660c80752a652b40
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000002500.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:709ac725be6cf86a1aadbfa013a7735f5b9b2f1bb78a16a828e00c1e19ca1263
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000003000.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a615a1add2b648cadeb891b65b47e4df34f1ef2cfca21f6933747704cd8e3b11
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000003500.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d2e6a32ca9bd169d3907608d8624415d7bc293f82bebe6767e5a2c1a74149607
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000004000.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3b2af9ec5826aaa0ac9b4b961741674402f25e6900eaa04f21231c2b7e71f252
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000004500.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:be6acb1b3a7c4b0dd72e2585e8e64de0115c568c5bc1b316a7925f97124d09ba
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/dlay_flux2_loha_1gpu_v2_000005000.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:385a943012296a55ebb1b73465bb4ee75891d28c2d10379baf675c19877a4799
|
| 3 |
+
size 3120664432
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963831954__000000500_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963885253__000000500_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764963937896__000000500_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966287976__000001000_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966340638__000001000_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764966393280__000001000_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968750269__000001500_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968802942__000001500_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764968855619__000001500_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971191622__000002000_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971244311__000002000_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764971297007__000002000_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973637152__000002500_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973689848__000002500_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764973742557__000002500_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976089753__000003000_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976142455__000003000_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764976195153__000003000_2-checkpoint.jpg
ADDED
|
Git LFS Details
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978551151__000003500_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978603817__000003500_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764978656496__000003500_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764980981509__000004000_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764981034171__000004000_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764981086832__000004000_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983432333__000004500_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983484988__000004500_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764983537656__000004500_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985882360__000005000_0-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985935041__000005000_1-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/.ipynb_checkpoints/1764985987725__000005000_2-checkpoint.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764963831954__000000500_0.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764963885253__000000500_1.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764963937896__000000500_2.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764966287976__000001000_0.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764966340638__000001000_1.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764966393280__000001000_2.jpg
ADDED
|
dlay_flux2_loha_1gpu_v2/samples/1764968750269__000001500_0.jpg
ADDED
|