Upload model via CLI

.gitattributes

@@ -33,3 +33,6 @@ 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/1752240289457__000001000_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1752242382071__000001500_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1752250836019__000003500_2.jpg filter=lfs diff=lfs merge=lfs -text

.job_config.json

@@ -0,0 +1,114 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "watch_raga",
+    "process": [
+      {
+        "type": "ui_trainer",
+        "training_folder": "/workspace/ai-toolkit/output",
+        "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
+        "device": "cuda",
+        "trigger_word": "r@ga",
+        "performance_log_every": 10,
+        "network": {
+          "type": "lora",
+          "linear": 64,
+          "linear_alpha": 64,
+          "conv": 16,
+          "conv_alpha": 16,
+          "lokr_full_rank": true,
+          "lokr_factor": -1,
+          "network_kwargs": {
+            "ignore_if_contains": []
+          }
+        },
+        "save": {
+          "dtype": "bf16",
+          "save_every": 500,
+          "max_step_saves_to_keep": 8,
+          "save_format": "diffusers",
+          "push_to_hub": false
+        },
+        "datasets": [
+          {
+            "folder_path": "/workspace/ai-toolkit/datasets/watch_raga",
+            "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": []
+          }
+        ],
+        "train": {
+          "batch_size": 1,
+          "bypass_guidance_embedding": false,
+          "steps": 4000,
+          "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,
+          "lr": 0.0004,
+          "ema_config": {
+            "use_ema": false,
+            "ema_decay": 0.99
+          },
+          "dtype": "bf16",
+          "diff_output_preservation": false,
+          "diff_output_preservation_multiplier": 1,
+          "diff_output_preservation_class": "person"
+        },
+        "model": {
+          "name_or_path": "black-forest-labs/FLUX.1-dev",
+          "quantize": false,
+          "quantize_te": true,
+          "arch": "flux",
+          "low_vram": false,
+          "model_kwargs": {}
+        },
+        "sample": {
+          "sampler": "flowmatch",
+          "sample_every": 500,
+          "width": 1024,
+          "height": 1024,
+          "prompts": [
+            "A photo-realistic still life shoot from a close-up angle about a luxury Rolex watch on a wooden surface with a reflective surface, set against a dark and moody background with soft, warm lighting. the image also shows a gold Rolex Oyster Perpetual Datejust watch with a polished finish and a gold bracelet. on the right side of the image, the watch is placed on a dark surface, with its reflection visible on the wooden surface. the watch has a gold dial with luminous hands and hour markers, and a date window at the 3 o'clock position. the background is blurred, with a subtle, warm light illuminating the watch, creating a sense of depth and dimensionality. the overall mood is sophisticated and sophisticated, with the focus on the watch and its reflection in the dark surface.",
+            "A close-up shot of a yellow rose, a gold watch, and a diamond necklace placed on a smooth, silky fabric background. the watch has a rectangular face with a white dial, and the necklace has a small diamond pendant in the shape of a butterfly. the yellow rose is placed next to the watch, creating a beautiful contrast between the two elements. the image is taken from a top-down perspective, focusing on the delicate details of the jewelry and the soft fabric texture of the fabric. the overall aesthetic is elegant and sophisticated, with a focus on the beauty of the items.",
+            "A close-up shot of a luxurious gold bracelet with white mother-of-pearl and garnet stones, showcasing intricate designs on the face of the watch. the watch has a rectangular shape with a gold case and a white mother of pearl dial, adorned with red gemstones. the bracelet is intricately designed with a floral pattern and has a polished finish, reflecting the light and creating a shimmering effect on its surface. the background is blurred, but it appears to be a neutral-colored surface, providing a subtle contrast to the intricate details of the bracelet. the overall aesthetic of the image is elegant and sophisticated, making it a perfect accessory for any special occasion.",
+            "A close-up shot of a minimalistic coffee setting with a cup and saucer on a dark surface. the focus is selectively placed on the watch, which has a black face with gold accents and a braided gold chain link bracelet. the watch is positioned next to a white coffee cup on the saucer, creating a subtle contrast between the two colors. the background is blurred, giving the focus to the watch and the cup. the overall aesthetic of the image is minimalistic and elegant, with a focus on the minimalist design of the watch."
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 3.5,
+          "sample_steps": 30,
+          "num_frames": 1,
+          "fps": 1
+        }
+      }
+    ]
+  },
+  "meta": {
+    "name": "[name]",
+    "version": "1.0"
+  }
+}

README.md

@@ -0,0 +1,87 @@
+---
+tags:
+- text-to-image
+- flux
+- lora
+- diffusers
+- template:sd-lora
+- ai-toolkit
+base_model: black-forest-labs/FLUX.1-dev
+instance_prompt: r@ga
+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: A photo-realistic still life shoot from a close-up angle about a luxury Rolex
+    watch on a wooden surface with a reflective surface, set against a dark and moody
+    background with soft, warm lighting. the image also shows a gold Rolex Oyster
+    Perpetual Datejust watch with a polished finish and a gold bracelet. on the right
+    side of the image, the watch is placed on a dark surface, with its reflection
+    visible on the wooden surface. the watch has a gold dial with luminous hands and
+    hour markers, and a date window at the 3 o'clock position. the background is blurred,
+    with a subtle, warm light illuminating the watch, creating a sense of depth and
+    dimensionality. the overall mood is sophisticated and sophisticated, with the
+    focus on the watch and its reflection in the dark surface.
+  output:
+    url: samples/1752252956803__000004000_3.jpg
+- text: A close-up shot of a yellow rose, a gold watch, and a diamond necklace placed
+    on a smooth, silky fabric background. the watch has a rectangular face with a
+    white dial, and the necklace has a small diamond pendant in the shape of a butterfly.
+    the yellow rose is placed next to the watch, creating a beautiful contrast between
+    the two elements. the image is taken from a top-down perspective, focusing on
+    the delicate details of the jewelry and the soft fabric texture of the fabric.
+    the overall aesthetic is elegant and sophisticated, with a focus on the beauty
+    of the items.
+  output:
+    url: samples/1752252938653__000004000_2.jpg
+- text: A close-up shot of a luxurious gold bracelet with white mother-of-pearl and
+    garnet stones, showcasing intricate designs on the face of the watch. the watch
+    has a rectangular shape with a gold case and a white mother of pearl dial, adorned
+    with red gemstones. the bracelet is intricately designed with a floral pattern
+    and has a polished finish, reflecting the light and creating a shimmering effect
+    on its surface. the background is blurred, but it appears to be a neutral-colored
+    surface, providing a subtle contrast to the intricate details of the bracelet.
+    the overall aesthetic of the image is elegant and sophisticated, making it a perfect
+    accessory for any special occasion.
+  output:
+    url: samples/1752252920460__000004000_1.jpg
+- text: A close-up shot of a minimalistic coffee setting with a cup and saucer on
+    a dark surface. the focus is selectively placed on the watch, which has a black
+    face with gold accents and a braided gold chain link bracelet. the watch is positioned
+    next to a white coffee cup on the saucer, creating a subtle contrast between the
+    two colors. the background is blurred, giving the focus to the watch and the cup.
+    the overall aesthetic of the image is minimalistic and elegant, with a focus on
+    the minimalist design of the watch.
+  output:
+    url: samples/1752252902308__000004000_0.jpg
+---
+
+# watch_raga
+
+Model trained with AI Toolkit by Ostris
+
+<Gallery />
+
+## Trigger words
+
+You should use `r@ga` 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/watch_raga/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/watch_raga', weight_name='watch_raga.safetensors')
+image = pipeline('r@ga A photo-realistic still life shoot from a close-up angle about a luxury Rolex watch on a wooden surface with a reflective surface, set against a dark and moody background with soft, warm lighting. the image also shows a gold Rolex Oyster Perpetual Datejust watch with a polished finish and a gold bracelet. on the right side of the image, the watch is placed on a dark surface, with its reflection visible on the wooden surface. the watch has a gold dial with luminous hands and hour markers, and a date window at the 3 o'clock position. the background is blurred, with a subtle, warm light illuminating the watch, creating a sense of depth and dimensionality. the overall mood is sophisticated and sophisticated, with the focus on the watch and its reflection in the dark surface.').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,125 @@
+job: extension
+config:
+  name: watch_raga
+  process:
+  - type: ui_trainer
+    training_folder: /workspace/ai-toolkit/output
+    sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
+    device: cuda
+    trigger_word: r@ga
+    performance_log_every: 10
+    network:
+      type: lora
+      linear: 64
+      linear_alpha: 64
+      conv: 16
+      conv_alpha: 16
+      lokr_full_rank: true
+      lokr_factor: -1
+      network_kwargs:
+        ignore_if_contains: []
+    save:
+      dtype: bf16
+      save_every: 500
+      max_step_saves_to_keep: 8
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /workspace/ai-toolkit/datasets/watch_raga
+      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: []
+    train:
+      batch_size: 1
+      bypass_guidance_embedding: false
+      steps: 4000
+      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
+      lr: 0.0004
+      ema_config:
+        use_ema: false
+        ema_decay: 0.99
+      dtype: bf16
+      diff_output_preservation: false
+      diff_output_preservation_multiplier: 1
+      diff_output_preservation_class: person
+    model:
+      name_or_path: black-forest-labs/FLUX.1-dev
+      quantize: false
+      quantize_te: true
+      arch: flux
+      low_vram: false
+      model_kwargs: {}
+    sample:
+      sampler: flowmatch
+      sample_every: 500
+      width: 1024
+      height: 1024
+      prompts:
+      - A photo-realistic still life shoot from a close-up angle about a luxury Rolex
+        watch on a wooden surface with a reflective surface, set against a dark and
+        moody background with soft, warm lighting. the image also shows a gold Rolex
+        Oyster Perpetual Datejust watch with a polished finish and a gold bracelet.
+        on the right side of the image, the watch is placed on a dark surface, with
+        its reflection visible on the wooden surface. the watch has a gold dial with
+        luminous hands and hour markers, and a date window at the 3 o'clock position.
+        the background is blurred, with a subtle, warm light illuminating the watch,
+        creating a sense of depth and dimensionality. the overall mood is sophisticated
+        and sophisticated, with the focus on the watch and its reflection in the dark
+        surface.
+      - A close-up shot of a yellow rose, a gold watch, and a diamond necklace placed
+        on a smooth, silky fabric background. the watch has a rectangular face with
+        a white dial, and the necklace has a small diamond pendant in the shape of
+        a butterfly. the yellow rose is placed next to the watch, creating a beautiful
+        contrast between the two elements. the image is taken from a top-down perspective,
+        focusing on the delicate details of the jewelry and the soft fabric texture
+        of the fabric. the overall aesthetic is elegant and sophisticated, with a
+        focus on the beauty of the items.
+      - A close-up shot of a luxurious gold bracelet with white mother-of-pearl and
+        garnet stones, showcasing intricate designs on the face of the watch. the
+        watch has a rectangular shape with a gold case and a white mother of pearl
+        dial, adorned with red gemstones. the bracelet is intricately designed with
+        a floral pattern and has a polished finish, reflecting the light and creating
+        a shimmering effect on its surface. the background is blurred, but it appears
+        to be a neutral-colored surface, providing a subtle contrast to the intricate
+        details of the bracelet. the overall aesthetic of the image is elegant and
+        sophisticated, making it a perfect accessory for any special occasion.
+      - A close-up shot of a minimalistic coffee setting with a cup and saucer on
+        a dark surface. the focus is selectively placed on the watch, which has a
+        black face with gold accents and a braided gold chain link bracelet. the watch
+        is positioned next to a white coffee cup on the saucer, creating a subtle
+        contrast between the two colors. the background is blurred, giving the focus
+        to the watch and the cup. the overall aesthetic of the image is minimalistic
+        and elegant, with a focus on the minimalist design of the watch.
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 3.5
+      sample_steps: 30
+      num_frames: 1
+      fps: 1
+meta:
+  name: watch_raga
+  version: '1.0'

optimizer.pt

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

watch_raga.safetensors

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

watch_raga_000000500.safetensors

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

watch_raga_000001000.safetensors

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

watch_raga_000001500.safetensors

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

watch_raga_000002000.safetensors

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

watch_raga_000002500.safetensors

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

watch_raga_000003000.safetensors

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

watch_raga_000003500.safetensors

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