Uploading full model folder

.gitattributes

@@ -33,3 +33,39 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767264032324__000000500_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767264528154__000001000_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767264530549__000001000_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767265028176__000001500_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767265030551__000001500_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767265533832__000002000_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767265536351__000002000_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767266034240__000002500_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767266036687__000002500_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767266535756__000003000_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767266538178__000003000_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767267038786__000003500_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767267041188__000003500_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767267536183__000004000_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767267538596__000004000_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767268035243__000004500_0-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/.ipynb_checkpoints/1767268037655__000004500_1-checkpoint.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767264032324__000000500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767264528154__000001000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767264530549__000001000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767265028176__000001500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767265030551__000001500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767265533832__000002000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767265536351__000002000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767266034240__000002500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767266036687__000002500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767266535756__000003000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767266538178__000003000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767267038786__000003500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767267041188__000003500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767267536183__000004000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767267538596__000004000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767268035243__000004500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767268037655__000004500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767268531513__000005000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1767268534030__000005000_1.jpg filter=lfs diff=lfs merge=lfs -text

.ipynb_checkpoints/config-checkpoint.yaml

@@ -0,0 +1,112 @@
+job: extension
+config:
+  name: ginger_zimage_turbo_lora
+  process:
+  - type: diffusion_trainer
+    training_folder: /out
+    sqlite_db_path: ./aitk_db.db
+    device: cuda
+    performance_log_every: 10
+    network:
+      type: lora
+      linear: 32
+      linear_alpha: 32
+      conv: 16
+      conv_alpha: 16
+      lokr_full_rank: true
+      lokr_factor: -1
+      network_kwargs:
+        ignore_if_contains: []
+    save:
+      dtype: bf16
+      save_every: 500
+      max_step_saves_to_keep: 10
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /ginger_dataset
+      mask_path: null
+      mask_min_value: 0.1
+      default_caption: ''
+      caption_ext: txt
+      caption_dropout_rate: 0.05
+      cache_latents_to_disk: false
+      is_reg: false
+      network_weight: 1
+      resolution:
+      - 1024
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      do_i2v: true
+      flip_x: false
+      flip_y: false
+    train:
+      batch_size: 1
+      bypass_guidance_embedding: false
+      steps: 5000
+      gradient_accumulation: 1
+      train_unet: true
+      train_text_encoder: false
+      gradient_checkpointing: false
+      noise_scheduler: flowmatch
+      optimizer: adamw8bit
+      timestep_type: weighted
+      content_or_style: balanced
+      optimizer_params:
+        weight_decay: 0.0001
+      unload_text_encoder: false
+      cache_text_embeddings: true
+      lr: 0.0001
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      skip_first_sample: true
+      force_first_sample: false
+      disable_sampling: false
+      dtype: bf16
+      diff_output_preservation: false
+      diff_output_preservation_multiplier: 1
+      diff_output_preservation_class: person
+      switch_boundary_every: 1
+      loss_type: mse
+      do_differential_guidance: true
+      differential_guidance_scale: 3
+    model:
+      name_or_path: Tongyi-MAI/Z-Image-Turbo
+      quantize: false
+      qtype: qfloat8
+      quantize_te: false
+      qtype_te: qfloat8
+      arch: zimage:turbo
+      low_vram: false
+      model_kwargs: {}
+      layer_offloading: false
+      layer_offloading_text_encoder_percent: 1
+      layer_offloading_transformer_percent: 1
+      assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
+    sample:
+      sampler: flowmatch
+      sample_every: 500
+      width: 896
+      height: 1152
+      samples:
+      - prompt: "ohwx woman stands confidently under a black umbrella by the seaside,\
+          \ wearing a bold red-and-white striped bikini with gold hardware accents.\
+          \ Her sun-kissed skin glows beneath bright daylight as she shields her eyes\
+          \ while holding a pink cocktail\u2014her long, flowing strawberry-blonde\
+          \ hair catching the breeze. With poised elegance and a serene gaze, she\
+          \ exudes relaxed luxury against the backdrop of ocean waves and beach loungers.\
+          \ The warm light enhances her curvaceous silhouette, creating an inviting\
+          \ atmosphere of tropical indulgence and effortless charm."
+      - prompt: a ohwx woman holding a coffee cup, sitting at a cafe
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 1
+      sample_steps: 8
+      num_frames: 1
+      fps: 1
+meta:
+  name: ginger_zimage_turbo_lora
+  version: '1.0'

config.yaml

@@ -0,0 +1,112 @@
+job: extension
+config:
+  name: ginger_zimage_turbo_lora
+  process:
+  - type: diffusion_trainer
+    training_folder: /out
+    sqlite_db_path: ./aitk_db.db
+    device: cuda
+    performance_log_every: 10
+    network:
+      type: lora
+      linear: 32
+      linear_alpha: 32
+      conv: 16
+      conv_alpha: 16
+      lokr_full_rank: true
+      lokr_factor: -1
+      network_kwargs:
+        ignore_if_contains: []
+    save:
+      dtype: bf16
+      save_every: 500
+      max_step_saves_to_keep: 10
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /ginger_dataset
+      mask_path: null
+      mask_min_value: 0.1
+      default_caption: ''
+      caption_ext: txt
+      caption_dropout_rate: 0.05
+      cache_latents_to_disk: false
+      is_reg: false
+      network_weight: 1
+      resolution:
+      - 1024
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      do_i2v: true
+      flip_x: false
+      flip_y: false
+    train:
+      batch_size: 1
+      bypass_guidance_embedding: false
+      steps: 5000
+      gradient_accumulation: 1
+      train_unet: true
+      train_text_encoder: false
+      gradient_checkpointing: false
+      noise_scheduler: flowmatch
+      optimizer: adamw8bit
+      timestep_type: weighted
+      content_or_style: balanced
+      optimizer_params:
+        weight_decay: 0.0001
+      unload_text_encoder: false
+      cache_text_embeddings: true
+      lr: 0.0001
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      skip_first_sample: true
+      force_first_sample: false
+      disable_sampling: false
+      dtype: bf16
+      diff_output_preservation: false
+      diff_output_preservation_multiplier: 1
+      diff_output_preservation_class: person
+      switch_boundary_every: 1
+      loss_type: mse
+      do_differential_guidance: true
+      differential_guidance_scale: 3
+    model:
+      name_or_path: Tongyi-MAI/Z-Image-Turbo
+      quantize: false
+      qtype: qfloat8
+      quantize_te: false
+      qtype_te: qfloat8
+      arch: zimage:turbo
+      low_vram: false
+      model_kwargs: {}
+      layer_offloading: false
+      layer_offloading_text_encoder_percent: 1
+      layer_offloading_transformer_percent: 1
+      assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
+    sample:
+      sampler: flowmatch
+      sample_every: 500
+      width: 896
+      height: 1152
+      samples:
+      - prompt: "ohwx woman stands confidently under a black umbrella by the seaside,\
+          \ wearing a bold red-and-white striped bikini with gold hardware accents.\
+          \ Her sun-kissed skin glows beneath bright daylight as she shields her eyes\
+          \ while holding a pink cocktail\u2014her long, flowing strawberry-blonde\
+          \ hair catching the breeze. With poised elegance and a serene gaze, she\
+          \ exudes relaxed luxury against the backdrop of ocean waves and beach loungers.\
+          \ The warm light enhances her curvaceous silhouette, creating an inviting\
+          \ atmosphere of tropical indulgence and effortless charm."
+      - prompt: a ohwx woman holding a coffee cup, sitting at a cafe
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 1
+      sample_steps: 8
+      num_frames: 1
+      fps: 1
+meta:
+  name: ginger_zimage_turbo_lora
+  version: '1.0'

ginger_zimage_turbo_lora.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:403d4bb91f2695e6eb92a133336bc23e8e08fd0a6648cca1739270105fe91390
+size 170128248

ginger_zimage_turbo_lora_000000500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82e3ed20b15bed85cc16411ca1fa1c65a5eb641ce1d6c97a5c987f8310425b9e
+size 170128248

ginger_zimage_turbo_lora_000001000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86e60fd5d039ac7df85e31cd2e5fb6065e73f01a2e403371e12721b35db74f0e
+size 170128248

ginger_zimage_turbo_lora_000001500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d1e47dbb3a48303de9a90244bc60bf322e14c75c24727474bc6a930f5a7659e
+size 170128248

ginger_zimage_turbo_lora_000002000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:296b49ef74f82d26f2c164d4b4455a2bea66c1ec2c2cea2b073d80c627750beb
+size 170128248

ginger_zimage_turbo_lora_000002500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:891fc5c6308555f42fd4997cfdd9ad1ee059d766a1abbc300f72ca69d216a47e
+size 170128248

ginger_zimage_turbo_lora_000003000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:393e8eb98986933bc026fe6416b499af2da492207942df1bf63aa2a1b7b5ce05
+size 170128248

ginger_zimage_turbo_lora_000003500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d671b9e1e50850ce23850267c4c3d394dc71cb675c0eb027f0a2ca56e8358106
+size 170128248

ginger_zimage_turbo_lora_000004000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b4896496a0c46f0301a706ef987fabc9a182074beffc26bf4e1dfcb0dbc9dbb
+size 170128248

ginger_zimage_turbo_lora_000004500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6ddc51ecd569303344831ac9baaf6389e01e5b15062bdf4d3177f542a26b7ec
+size 170128248

optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46f3ccaa3f27c736ec27852d95f3386232b65107c0a0a87a2078e217c3ed171c
+size 173288293