Upload model via CLI

.job_config.json

@@ -0,0 +1,131 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "W_sunglass",
+    "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,
+              1536,
+              1280
+            ],
+            "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": "sigmoid",
+          "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 in close-up front portrait pose wearing narrow oval black sunglasses, loose wavy hair, dressed in a deep maroon open-collar blouse, natural confident expression"
+            },
+            {
+              "prompt": "s@ngl@ss woman portrait pose, wearing round gradient brown lens sunglasses with thin golden bridge and brown acetate frame, styled with natural wavy dark hair, minimal natural makeup, lips in nude tone, dressed in a strappy outfit with tortoiseshell detail, pose with right hand gently adjusting the sunglasses, close-up shot, soft daylight background, "
+            },
+            {
+              "prompt": "s@ngl@ss woman close-up seated portrait pose, resting chin lightly on hand, short curly hair, wearing amber-brown wayfarer style sunglasses with dark lenses, draped tan off-shoulder outfit, metallic choker necklace, warm studio backdrop enhancing elegance and artistic mood"
+            },
+            {
+              "prompt": "s@ngl@ss woman portrait pose, three-quarter side view, wearing rectangular tortoiseshell sunglasses with bold temples featuring gold “MIU MIU” branding, hair tied back in a sleek low ponytail, natural minimal makeup, dressed in a pale blue collared shirt, neutral white background,"
+            }
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 2.8,
+          "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/1755933382510__000000000_0.jpg
+- text: Sample generation
+  output:
+    url: samples/1755933400301__000000000_1.jpg
+- text: Sample generation
+  output:
+    url: samples/1755933418473__000000000_2.jpg
+---
+
+# W_sunglass
+
+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/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', weight_name='W_sunglass_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.safetensors

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

W_sunglass_000000250.safetensors

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

W_sunglass_000000500.safetensors

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

W_sunglass_000000750.safetensors

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

W_sunglass_000001000.safetensors

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

W_sunglass_000001250.safetensors

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

W_sunglass_000001500.safetensors

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

W_sunglass_000001750.safetensors

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

config.yaml

@@ -0,0 +1,115 @@
+job: extension
+config:
+  name: W_sunglass
+  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
+      - 1536
+      - 1280
+      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: sigmoid
+      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 in close-up front portrait pose wearing narrow oval
+          black sunglasses, loose wavy hair, dressed in a deep maroon open-collar
+          blouse, natural confident expression
+      - prompt: 's@ngl@ss woman portrait pose, wearing round gradient brown lens sunglasses
+          with thin golden bridge and brown acetate frame, styled with natural wavy
+          dark hair, minimal natural makeup, lips in nude tone, dressed in a strappy
+          outfit with tortoiseshell detail, pose with right hand gently adjusting
+          the sunglasses, close-up shot, soft daylight background, '
+      - prompt: s@ngl@ss woman close-up seated portrait pose, resting chin lightly
+          on hand, short curly hair, wearing amber-brown wayfarer style sunglasses
+          with dark lenses, draped tan off-shoulder outfit, metallic choker necklace,
+          warm studio backdrop enhancing elegance and artistic mood
+      - prompt: "s@ngl@ss woman portrait pose, three-quarter side view, wearing rectangular\
+          \ tortoiseshell sunglasses with bold temples featuring gold \u201CMIU MIU\u201D\
+          \ branding, hair tied back in a sleek low ponytail, natural minimal makeup,\
+          \ dressed in a pale blue collared shirt, neutral white background,"
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 2.8
+      sample_steps: 30
+      num_frames: 1
+      fps: 1
+meta:
+  name: W_sunglass
+  version: '1.0'

optimizer.pt

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