LoRA + generated artifacts (3000 steps)

.gitattributes

@@ -37,3 +37,60 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 1771963478038__000000000_1.jpg filter=lfs diff=lfs merge=lfs -text
 1771963517447__000000000_2.jpg filter=lfs diff=lfs merge=lfs -text
 1771963556789__000000000_3.jpg filter=lfs diff=lfs merge=lfs -text
+loss_log.db filter=lfs diff=lfs merge=lfs -text
+loss_log.db-wal filter=lfs diff=lfs merge=lfs -text
+samples/1772473243753__000000000_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772473398442__000000000_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475443233__000000250_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475481907__000000250_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475597966__000000250_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475636704__000000250_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475675413__000000250_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772475791581__000000250_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772477960072__000000500_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772477999099__000000500_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772478038181__000000500_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772480414531__000000750_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772480453308__000000750_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772482616868__000001000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772482771492__000001000_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772482848776__000001000_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772484928694__000001250_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772484967445__000001250_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772485083519__000001250_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772485122221__000001250_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772485160979__000001250_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772485277085__000001250_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487352662__000001500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487391401__000001500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487507555__000001500_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487546280__000001500_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487585026__000001500_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772487701231__000001500_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772489773532__000001750_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772489812191__000001750_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772489850836__000001750_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772489928209__000001750_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772489966905__000001750_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772490005586__000001750_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772490121563__000001750_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492140468__000002000_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492179180__000002000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492217895__000002000_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492295299__000002000_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492333978__000002000_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492372657__000002000_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772492488691__000002000_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772494640384__000002250_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772494679106__000002250_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772494717823__000002250_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772494833979__000002250_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772497108079__000002500_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772497146920__000002500_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772497263282__000002500_9.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499299914__000002750_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499338719__000002750_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499454928__000002750_4.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499493692__000002750_5.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499532491__000002750_6.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1772499648849__000002750_9.jpg filter=lfs diff=lfs merge=lfs -text

Flat_Design_Motion_Graphics_000002000.safetensors

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

Flat_Design_Motion_Graphics_000002250.safetensors

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

Flat_Design_Motion_Graphics_000002500.safetensors

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

Flat_Design_Motion_Graphics_000002750.safetensors

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

config.yaml

@@ -0,0 +1,127 @@
+job: extension
+config:
+  name: Flat_Design_Motion_Graphics
+  process:
+  - type: diffusion_trainer
+    training_folder: /app/ai-toolkit/output
+    sqlite_db_path: ./aitk_db.db
+    device: cuda
+    trigger_word: Flat_Design_Motion_Graphics_Style
+    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: 250
+      max_step_saves_to_keep: 4
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /app/ai-toolkit/datasets/flat_design_motion_graphics
+      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:
+      - 512
+      - 768
+      - 1024
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      flip_x: false
+      flip_y: false
+      num_repeats: 1
+    train:
+      batch_size: 1
+      bypass_guidance_embedding: false
+      steps: 3000
+      gradient_accumulation: 1
+      train_unet: true
+      train_text_encoder: false
+      gradient_checkpointing: true
+      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: false
+      lr: 0.0001
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      skip_first_sample: false
+      force_first_sample: false
+      disable_sampling: false
+      dtype: bf16
+      diff_output_preservation: true
+      diff_output_preservation_multiplier: 1
+      diff_output_preservation_class: image
+      switch_boundary_every: 1
+      loss_type: mse
+    logging:
+      log_every: 1
+      use_ui_logger: true
+    model:
+      name_or_path: black-forest-labs/FLUX.2-dev
+      quantize: true
+      qtype: qfloat8
+      quantize_te: true
+      qtype_te: qfloat8
+      arch: flux2
+      low_vram: true
+      model_kwargs:
+        match_target_res: true
+      layer_offloading: false
+      layer_offloading_text_encoder_percent: 1
+      layer_offloading_transformer_percent: 1
+    sample:
+      sampler: flowmatch
+      sample_every: 250
+      width: 1024
+      height: 1024
+      samples:
+      - prompt: Flat_Design_Motion_Graphics_Style woman with red hair, playing chess
+          at the park, bomb going off in the background
+      - prompt: Flat_Design_Motion_Graphics_Style woman holding a coffee cup, in a
+          beanie, sitting at a cafe
+      - prompt: Flat_Design_Motion_Graphics_Style horse is a DJ at a night club, fish
+          eye lens, smoke machine, lazer lights, holding a martini
+      - prompt: Flat_Design_Motion_Graphics_Style man showing off his cool new t shirt
+          at the beach, a shark is jumping out of the water in the background
+      - prompt: Flat_Design_Motion_Graphics_Style bear building a log cabin in the
+          snow covered mountains
+      - prompt: Flat_Design_Motion_Graphics_Style woman playing the guitar, on stage,
+          singing a song, laser lights, punk rocker
+      - prompt: Flat_Design_Motion_Graphics_Style hipster man with a beard, building
+          a chair, in a wood shop
+      - prompt: Flat_Design_Motion_Graphics_Style photo of a man, white background,
+          medium shot, modeling clothing, studio lighting, white backdrop
+      - prompt: Flat_Design_Motion_Graphics_Style a man holding a sign that says,
+          'this is a sign'
+      - prompt: Flat_Design_Motion_Graphics_Style a bulldog, in a post apocalyptic
+          world, with a shotgun, in a leather jacket, in a desert, with a motorcycle
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 4
+      sample_steps: 25
+      num_frames: 1
+      fps: 1
+  meta:
+    name: Flat_Design_Motion_Graphics
+    version: '1.0'

loss_log.db

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

loss_log.db-wal

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