Upload model via CLI

.job_config.json

@@ -0,0 +1,133 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "M_Sunglasses",
+    "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/m_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,
+            "flip_x": false,
+            "flip_y": false
+          }
+        ],
+        "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.0001,
+          "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,Translucent ice-blue frames rest on a soft grey surface, with streaks of diffused daylight creating fluid shadows around them. The clean backdrop and subtle reflections emphasize the modern, sculptural design of the eyewear."
+            },
+            {
+              "prompt": "s@ngl@ss, A pair of tortoiseshell square-frame glasses is carefully balanced upright against a warm beige surface, lit dramatically from the left. The angled spotlight casts elongated shadows, emphasizing the sculpted frame edges and polished acetate finish. The smooth gradient background adds depth, allowing the rich tones of the eyewear to stand out with timeless sophistication."
+            },
+            {
+              "prompt": "s@ngl@ss, A pair of cream-toned cat-eye sunglasses rests on a rippled satin backdrop, softly illuminated by warm daylight from the right. The flowing fabric casts delicate shadows that echo the curves of the frame, creating a luxurious interplay of texture and shine. The angled composition highlights the bold silhouette while the surface reflections on the lenses add depth and sophistication."
+            },
+            {
+              "prompt": "s@ngl@ss, Resting on a warm golden-beige fabric background, the cat-eye sunglasses take center stage with bold tortoiseshell frames and ivory temples. Soft directional lighting from the upper right casts elegant flowing shadows across the surface, adding depth and movement that accentuate the glossy lenses. The composition feels luxurious and cinematic, with the textured backdrop enhancing the sculptural form of the eyewear."
+            }
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 3,
+          "sample_steps": 30,
+          "num_frames": 1,
+          "fps": 1
+        }
+      }
+    ]
+  },
+  "meta": {
+    "name": "[name]",
+    "version": "1.0"
+  }
+}

M_Sunglasses.safetensors

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

M_Sunglasses_000000250.safetensors

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

M_Sunglasses_000000500.safetensors

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

M_Sunglasses_000000750.safetensors

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

M_Sunglasses_000001000.safetensors

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

M_Sunglasses_000001250.safetensors

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

M_Sunglasses_000001500.safetensors

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

M_Sunglasses_000001750.safetensors

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

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/1756188753236__000000000_0.jpg
+- text: Sample generation
+  output:
+    url: samples/1756188770813__000000000_1.jpg
+- text: Sample generation
+  output:
+    url: samples/1756188788435__000000000_2.jpg
+---
+
+# M_Sunglasses
+
+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/M_Sunglasses/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/M_Sunglasses', weight_name='M_Sunglasses_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)

config.yaml

@@ -0,0 +1,123 @@
+job: extension
+config:
+  name: M_Sunglasses
+  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/m_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
+      flip_x: false
+      flip_y: false
+    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.0001
+      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,Translucent ice-blue frames rest on a soft grey surface,
+          with streaks of diffused daylight creating fluid shadows around them. The
+          clean backdrop and subtle reflections emphasize the modern, sculptural design
+          of the eyewear.
+      - prompt: s@ngl@ss, A pair of tortoiseshell square-frame glasses is carefully
+          balanced upright against a warm beige surface, lit dramatically from the
+          left. The angled spotlight casts elongated shadows, emphasizing the sculpted
+          frame edges and polished acetate finish. The smooth gradient background
+          adds depth, allowing the rich tones of the eyewear to stand out with timeless
+          sophistication.
+      - prompt: s@ngl@ss, A pair of cream-toned cat-eye sunglasses rests on a rippled
+          satin backdrop, softly illuminated by warm daylight from the right. The
+          flowing fabric casts delicate shadows that echo the curves of the frame,
+          creating a luxurious interplay of texture and shine. The angled composition
+          highlights the bold silhouette while the surface reflections on the lenses
+          add depth and sophistication.
+      - prompt: s@ngl@ss, Resting on a warm golden-beige fabric background, the cat-eye
+          sunglasses take center stage with bold tortoiseshell frames and ivory temples.
+          Soft directional lighting from the upper right casts elegant flowing shadows
+          across the surface, adding depth and movement that accentuate the glossy
+          lenses. The composition feels luxurious and cinematic, with the textured
+          backdrop enhancing the sculptural form of the eyewear.
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 3
+      sample_steps: 30
+      num_frames: 1
+      fps: 1
+meta:
+  name: M_Sunglasses
+  version: '1.0'

optimizer.pt

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