Upload final Shantae Krea 2 Turbo LoRA and samples

.gitattributes

@@ -33,3 +33,5 @@ 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/sample_step3000_02.jpg filter=lfs diff=lfs merge=lfs -text
+samples/sample_step3000_09.jpg filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,61 @@
+---
+library_name: diffusers
+pipeline_tag: text-to-image
+base_model: krea/Krea-2-Turbo
+tags:
+  - lora
+  - krea-2-turbo
+  - ai-toolkit
+  - text-to-image
+  - portrait
+---
+
+# Shantae Krea 2 Turbo LoRA v1
+
+Identity LoRA trained for **Krea 2 Turbo** with Ostris AI Toolkit.
+
+## Usage
+
+- **Trigger word:** `shantae`
+- **Base model:** `krea/Krea-2-Turbo`
+- **LoRA file:** [`shantae_krea2_turbo_v1_step3000.safetensors`](./shantae_krea2_turbo_v1_step3000.safetensors)
+- **Recommended starting point:** 9 steps, guidance 1, LoRA strength 0.8–1.0
+
+Example prompt:
+
+```text
+shantae, close-up portrait, natural window light, detailed skin, shallow depth of field
+```
+
+## Sample images
+
+All samples below were generated from the final step-3000 checkpoint at 1024×1024.
+
+| | |
+|---|---|
+| ![Close-up portrait](samples/sample_step3000_01.jpg) | ![Golden-hour portrait](samples/sample_step3000_02.jpg) |
+| ![City fashion portrait](samples/sample_step3000_03.jpg) | ![Cafe portrait](samples/sample_step3000_04.jpg) |
+| ![Studio portrait](samples/sample_step3000_05.jpg) | ![Beach portrait](samples/sample_step3000_06.jpg) |
+| ![Rooftop evening portrait](samples/sample_step3000_07.jpg) | ![Cozy apartment portrait](samples/sample_step3000_08.jpg) |
+| ![Park portrait](samples/sample_step3000_09.jpg) | ![Neon city portrait](samples/sample_step3000_10.jpg) |
+
+## Training details
+
+| Setting | Value |
+|---|---|
+| Training steps | 3,000 |
+| LoRA rank / alpha | 32 / 32 |
+| Precision | BF16 |
+| Optimizer | AdamW8Bit |
+| Learning rate | 0.0001 |
+| Batch size | 1 |
+| Gradient accumulation | 1 |
+| Resolution buckets | 512, 768, 1024 |
+| Scheduler | FlowMatch |
+| Sample steps / guidance | 9 / 1 |
+
+The original AI Toolkit configuration is included as [`config.yaml`](./config.yaml).
+
+## Notes
+
+Generated images may reproduce limitations or biases present in the training data and base model. Review outputs before use, especially in public or commercial contexts.

config.yaml

@@ -0,0 +1,126 @@
+job: extension
+meta:
+  name: shantae_krea2_turbo_v1
+  version: '1.0'
+config:
+  name: shantae_krea2_turbo_v1
+  process:
+  - save:
+      dtype: bf16
+      save_every: 250
+      push_to_hub: false
+      save_format: diffusers
+      max_step_saves_to_keep: 4
+    type: diffusion_trainer
+    model:
+      arch: krea2:turbo
+      qtype: qfloat8
+      compile: false
+      low_vram: true
+      qtype_te: qfloat8
+      quantize: true
+      quantize_te: true
+      model_kwargs: {}
+      name_or_path: krea/Krea-2-Turbo
+      layer_offloading: false
+      assistant_lora_path: ostris/krea2_turbo_training_adapter/krea2_turbo_training_adapter_v1.safetensors
+      layer_offloading_transformer_percent: 1
+      layer_offloading_text_encoder_percent: 1
+    train:
+      lr: 0.0001
+      dtype: bf16
+      steps: 3000
+      loss_type: mse
+      optimizer: adamw8bit
+      batch_size: 1
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      train_unet: true
+      timestep_type: linear
+      noise_scheduler: flowmatch
+      content_or_style: balanced
+      disable_sampling: false
+      optimizer_params:
+        weight_decay: 0.0001
+      skip_first_sample: false
+      force_first_sample: false
+      train_text_encoder: false
+      unload_text_encoder: false
+      cache_text_embeddings: false
+      gradient_accumulation: 1
+      switch_boundary_every: 1
+      gradient_checkpointing: true
+      diff_output_preservation: false
+      bypass_guidance_embedding: false
+      diff_output_preservation_class: person
+      diff_output_preservation_multiplier: 1
+    device: cuda
+    sample:
+      fps: 1
+      neg: ''
+      seed: 42
+      width: 1024
+      height: 1024
+      sampler: flowmatch
+      samples:
+      - prompt: shantae, close-up portrait, natural window light, detailed skin, shallow
+          depth of field
+      - prompt: shantae, waist-up portrait outdoors at golden hour, casual outfit,
+          cinematic photography
+      - prompt: shantae, full-body fashion portrait on a city street, confident pose,
+          soft daylight
+      - prompt: shantae, seated in a modern cafe, candid expression, warm ambient
+          lighting
+      - prompt: shantae, studio portrait against a clean neutral backdrop, professional
+          softbox lighting
+      - prompt: shantae, walking along a beach at sunset, wind in hair, natural candid
+          photograph
+      - prompt: shantae, elegant evening outfit at a rooftop venue, city lights bokeh,
+          editorial photo
+      - prompt: shantae, cozy sweater in a bright apartment, relaxed pose, lifestyle
+          photography
+      - prompt: shantae, athletic outfit in a park, dynamic pose, crisp morning light
+      - prompt: shantae, black leather jacket under neon city lights, cinematic night
+          portrait
+      walk_seed: true
+      num_frames: 1
+      sample_every: 500
+      sample_steps: 9
+      guidance_scale: 1
+      sample_start_step: 0
+    logging:
+      log_every: 1
+      use_ui_logger: true
+    network:
+      type: lora
+      linear: 32
+      lokr_factor: -1
+      linear_alpha: 32
+      lokr_full_rank: true
+      network_kwargs:
+        ignore_if_contains: []
+    datasets:
+    - flip_x: false
+      flip_y: false
+      is_reg: false
+      controls: []
+      mask_path: null
+      num_frames: 1
+      resolution:
+      - 512
+      - 768
+      - 1024
+      caption_ext: txt
+      folder_path: /app/ai-toolkit/datasets/shantae_v1
+      num_repeats: 1
+      mask_min_value: 0.1
+      network_weight: 1
+      default_caption: shantae
+      caption_dropout_rate: 0.05
+      cache_latents_to_disk: true
+      shrink_video_to_frames: true
+    trigger_word: shantae
+    sqlite_db_path: /app/ai-toolkit/aitk_db.db
+    training_folder: /app/ai-toolkit/output
+    performance_log_every: 10

shantae_krea2_turbo_v1_step3000.safetensors

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