Upload model via CLI

.gitattributes

@@ -33,3 +33,11 @@ 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/1755946363372__000000250_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755947196226__000000500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755948857455__000001000_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755950506980__000001500_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755950525092__000001500_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755951351829__000001750_0.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755951370140__000001750_1.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1755952220038__000002000_1.jpg filter=lfs diff=lfs merge=lfs -text

.job_config.json

@@ -0,0 +1,131 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "W_sunglass_Shift",
+    "process": [
+      {
+        "type": "ui_trainer",
+        "training_folder": "/workspace/ai-toolkit/output",
+        "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
+        "device": "cuda",
+        "trigger_word": "s@ngl@ss",
+        "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": 8,
+          "save_format": "diffusers",
+          "push_to_hub": false
+        },
+        "datasets": [
+          {
+            "folder_path": "/workspace/ai-toolkit/datasets/w_sunglass",
+            "control_path": null,
+            "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,
+              1280,
+              1536
+            ],
+            "controls": [],
+            "shrink_video_to_frames": true,
+            "num_frames": 1,
+            "do_i2v": true
+          }
+        ],
+        "train": {
+          "batch_size": 1,
+          "bypass_guidance_embedding": false,
+          "steps": 2000,
+          "gradient_accumulation": 1,
+          "train_unet": true,
+          "train_text_encoder": false,
+          "gradient_checkpointing": true,
+          "noise_scheduler": "flowmatch",
+          "optimizer": "adamw8bit",
+          "timestep_type": "shift",
+          "content_or_style": "balanced",
+          "optimizer_params": {
+            "weight_decay": 0.0001
+          },
+          "unload_text_encoder": false,
+          "cache_text_embeddings": false,
+          "lr": 0.0002,
+          "ema_config": {
+            "use_ema": true,
+            "ema_decay": 0.99
+          },
+          "skip_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
+        },
+        "model": {
+          "name_or_path": "black-forest-labs/FLUX.1-dev",
+          "quantize": false,
+          "qtype": "qfloat8",
+          "quantize_te": true,
+          "qtype_te": "qfloat8",
+          "arch": "flux",
+          "low_vram": false,
+          "model_kwargs": {}
+        },
+        "sample": {
+          "sampler": "flowmatch",
+          "sample_every": 250,
+          "width": 1024,
+          "height": 1024,
+          "samples": [
+            {
+              "prompt": "s@ngl@ss woman portrait pose, three-quarter side view looking back over shoulder, wearing square oversized tortoiseshell patterned sunglasses with dark brown lenses, natural curly hair flowing freely, minimal makeup with soft nude lips, dressed in white strap top, clean white background, s@ngl@ss"
+            },
+            {
+              "prompt": "s@ngl@ss woman portrait pose, front-facing view, wearing slim oval brown-tinted sunglasses with thin gold metallic frame, blonde loose waves styled naturally, minimal makeup with nude lips, accessorized with chunky gold chain necklace and hoop earrings, dressed in beige knit top, neutral background, s@ngl@ss"
+            },
+            {
+              "prompt": "s@ngl@ss woman portrait pose, three-quarter angle, wearing round gold-rimmed sunglasses with green lenses, minimal makeup, small hoop earrings, styled in straight black hair, dressed in a light blue chambray shirt layered over a white ribbed top, soft neutral studio background, s@ngl@ss"
+            },
+            {
+              "prompt": "s@ngl@ss woman portrait pose, seated three-quarter angle, wearing bold black rectangular sunglasses with chunky chain temples and gold emblem detail, hair sleekly tied back, minimal makeup with soft glow, dressed in a sleeveless black top, hand placed elegantly on shoulder, neutral studio background, s@ngl@ss"
+            }
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 3,
+          "sample_steps": 30,
+          "num_frames": 1,
+          "fps": 1
+        }
+      }
+    ]
+  },
+  "meta": {
+    "name": "[name]",
+    "version": "1.0"
+  }
+}

README.md

@@ -0,0 +1,54 @@
+---
+tags:
+- text-to-image
+- flux
+- lora
+- diffusers
+- template:sd-lora
+- ai-toolkit
+base_model: black-forest-labs/FLUX.1-dev
+instance_prompt: s@ngl@ss
+license: other
+license_name: flux-1-dev-non-commercial-license
+license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
+widget:
+- text: Sample generation
+  output:
+    url: samples/1755945497667__000000000_0.jpg
+- text: Sample generation
+  output:
+    url: samples/1755945515364__000000000_1.jpg
+- text: Sample generation
+  output:
+    url: samples/1755945533146__000000000_2.jpg
+---
+
+# W_sunglass_Shift
+
+Model trained with AI Toolkit by Ostris
+
+<Gallery />
+
+## Trigger words
+
+You should use `s@ngl@ss` to trigger the image generation.
+
+## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, etc.
+
+Weights for this model are available in Safetensors format.
+
+[Download](/username/W_sunglass_Shift/tree/main) them in the Files & versions tab.
+
+## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
+
+```py
+from diffusers import AutoPipelineForText2Image
+import torch
+
+pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
+pipeline.load_lora_weights('username/W_sunglass_Shift', weight_name='W_sunglass_Shift_000000250.safetensors')
+image = pipeline('s@ngl@ss style artwork').images[0]
+image.save("my_image.png")
+```
+
+For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)

W_sunglass_Shift.safetensors

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

W_sunglass_Shift_000000250.safetensors

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

W_sunglass_Shift_000000500.safetensors

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

W_sunglass_Shift_000000750.safetensors

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

W_sunglass_Shift_000001000.safetensors

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

W_sunglass_Shift_000001250.safetensors

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

W_sunglass_Shift_000001500.safetensors

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

W_sunglass_Shift_000001750.safetensors

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

config.yaml

@@ -0,0 +1,118 @@
+job: extension
+config:
+  name: W_sunglass_Shift
+  process:
+  - type: ui_trainer
+    training_folder: /workspace/ai-toolkit/output
+    sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
+    device: cuda
+    trigger_word: s@ngl@ss
+    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: 8
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /workspace/ai-toolkit/datasets/w_sunglass
+      control_path: null
+      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
+      - 1280
+      - 1536
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      do_i2v: true
+    train:
+      batch_size: 1
+      bypass_guidance_embedding: false
+      steps: 2000
+      gradient_accumulation: 1
+      train_unet: true
+      train_text_encoder: false
+      gradient_checkpointing: true
+      noise_scheduler: flowmatch
+      optimizer: adamw8bit
+      timestep_type: shift
+      content_or_style: balanced
+      optimizer_params:
+        weight_decay: 0.0001
+      unload_text_encoder: false
+      cache_text_embeddings: false
+      lr: 0.0002
+      ema_config:
+        use_ema: true
+        ema_decay: 0.99
+      skip_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
+    model:
+      name_or_path: black-forest-labs/FLUX.1-dev
+      quantize: false
+      qtype: qfloat8
+      quantize_te: true
+      qtype_te: qfloat8
+      arch: flux
+      low_vram: false
+      model_kwargs: {}
+    sample:
+      sampler: flowmatch
+      sample_every: 250
+      width: 1024
+      height: 1024
+      samples:
+      - prompt: s@ngl@ss woman portrait pose, three-quarter side view looking back
+          over shoulder, wearing square oversized tortoiseshell patterned sunglasses
+          with dark brown lenses, natural curly hair flowing freely, minimal makeup
+          with soft nude lips, dressed in white strap top, clean white background,
+          s@ngl@ss
+      - prompt: s@ngl@ss woman portrait pose, front-facing view, wearing slim oval
+          brown-tinted sunglasses with thin gold metallic frame, blonde loose waves
+          styled naturally, minimal makeup with nude lips, accessorized with chunky
+          gold chain necklace and hoop earrings, dressed in beige knit top, neutral
+          background, s@ngl@ss
+      - prompt: s@ngl@ss woman portrait pose, three-quarter angle, wearing round gold-rimmed
+          sunglasses with green lenses, minimal makeup, small hoop earrings, styled
+          in straight black hair, dressed in a light blue chambray shirt layered over
+          a white ribbed top, soft neutral studio background, s@ngl@ss
+      - prompt: s@ngl@ss woman portrait pose, seated three-quarter angle, wearing
+          bold black rectangular sunglasses with chunky chain temples and gold emblem
+          detail, hair sleekly tied back, minimal makeup with soft glow, dressed in
+          a sleeveless black top, hand placed elegantly on shoulder, neutral studio
+          background, s@ngl@ss
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 3
+      sample_steps: 30
+      num_frames: 1
+      fps: 1
+meta:
+  name: W_sunglass_Shift
+  version: '1.0'

optimizer.pt

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