Upload model via CLI

.gitattributes

@@ -33,3 +33,21 @@ 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/1756455407374__000000000_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756455422276__000000000_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756456232900__000000250_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756456248085__000000250_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756457060163__000000500_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756457075415__000000500_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756457894016__000000750_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756457909267__000000750_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756458720829__000001000_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756458736399__000001000_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756459543927__000001250_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756459559205__000001250_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756460383941__000001500_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756460399252__000001500_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756461219964__000001750_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756461235233__000001750_3.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756462025038__000002000_2.jpg filter=lfs diff=lfs merge=lfs -text
+samples/1756462040505__000002000_3.jpg filter=lfs diff=lfs merge=lfs -text

.job_config.json

@@ -0,0 +1,133 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "wallet-poses",
+    "process": [
+      {
+        "type": "ui_trainer",
+        "training_folder": "/workspace/ai-toolkit/output",
+        "sqlite_db_path": "/workspace/ai-toolkit/aitk_db.db",
+        "device": "cuda",
+        "trigger_word": "w@llet",
+        "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/wallet_poses",
+            "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": "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": "w@llet, Man in a beige blazer and white t-shirt, holding a black leather wallet close to his chin with his right hand, slightly angled to showcase the wallet’s texture and stitching. The composition is a tight portrait crop, centering on his face, upper torso, and wallet as the focal point. His gaze is directed downward with a subtle, thoughtful expression. The background is plain white with soft gradient lighting, evenly diffused to eliminate shadows, giving the scene a polished and minimal aesthetic that emphasizes the wallet’s premium finish."
+            },
+            {
+              "prompt": "w@llet, Man in a navy blazer and white t-shirt, holding a brown leather wallet with his right hand, placing it into the inner chest pocket of his blazer. The shot is a cropped mid-close composition, focusing on the torso and wallet, with the face partially visible. The background is plain light grey, softly lit with even daylight, ensuring no harsh shadows and keeping the focus on the subject and accessory."
+            },
+            {
+              "prompt": "w@llet, Close-up shot of a man’s right hand holding a brown textured leather wallet with a crocodile pattern, positioned just above the back pocket of blue denim jeans. His beige t-shirt is partially visible, creating a neutral backdrop. The wallet is angled slightly outward, emphasizing its grain, pattern, and stitched detailing. The composition highlights the casual styling and practical placement, with soft, diffused lighting enhancing the textures of both the leather and denim for a natural, lifestyle-oriented presentation."
+            },
+            {
+              "prompt": "w@llet,A man in a light grey suit jacket layered over a textured white knit sweater poses in a semi-formal look, holding a brown leather wallet in his right hand. His left hand adjusts the lapel of his jacket, giving the impression of placing the wallet into the inner pocket. The wallet is angled forward to showcase its grain texture and stitched edges. A sleek wristwatch peeks out on his left wrist, complementing the polished look. Background is a neutral soft grey with diffused studio lighting, creating a clean and professional aesthetic while keeping the focus on the wallet."
+            }
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 4,
+          "sample_steps": 25,
+          "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: w@llet
+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/1756455377818__000000000_0.jpg
+- text: Sample generation
+  output:
+    url: samples/1756455392563__000000000_1.jpg
+- text: Sample generation
+  output:
+    url: samples/1756455407374__000000000_2.jpg
+---
+
+# wallet-poses
+
+Model trained with AI Toolkit by Ostris
+
+<Gallery />
+
+## Trigger words
+
+You should use `w@llet` 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/wallet-poses/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/wallet-poses', weight_name='wallet-poses_000000250.safetensors')
+image = pipeline('w@llet 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,131 @@
+job: extension
+config:
+  name: wallet-poses
+  process:
+  - type: ui_trainer
+    training_folder: /workspace/ai-toolkit/output
+    sqlite_db_path: /workspace/ai-toolkit/aitk_db.db
+    device: cuda
+    trigger_word: w@llet
+    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/wallet_poses
+      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: 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: "w@llet, Man in a beige blazer and white t-shirt, holding a black\
+          \ leather wallet close to his chin with his right hand, slightly angled\
+          \ to showcase the wallet\u2019s texture and stitching. The composition is\
+          \ a tight portrait crop, centering on his face, upper torso, and wallet\
+          \ as the focal point. His gaze is directed downward with a subtle, thoughtful\
+          \ expression. The background is plain white with soft gradient lighting,\
+          \ evenly diffused to eliminate shadows, giving the scene a polished and\
+          \ minimal aesthetic that emphasizes the wallet\u2019s premium finish."
+      - prompt: w@llet, Man in a navy blazer and white t-shirt, holding a brown leather
+          wallet with his right hand, placing it into the inner chest pocket of his
+          blazer. The shot is a cropped mid-close composition, focusing on the torso
+          and wallet, with the face partially visible. The background is plain light
+          grey, softly lit with even daylight, ensuring no harsh shadows and keeping
+          the focus on the subject and accessory.
+      - prompt: "w@llet, Close-up shot of a man\u2019s right hand holding a brown\
+          \ textured leather wallet with a crocodile pattern, positioned just above\
+          \ the back pocket of blue denim jeans. His beige t-shirt is partially visible,\
+          \ creating a neutral backdrop. The wallet is angled slightly outward, emphasizing\
+          \ its grain, pattern, and stitched detailing. The composition highlights\
+          \ the casual styling and practical placement, with soft, diffused lighting\
+          \ enhancing the textures of both the leather and denim for a natural, lifestyle-oriented\
+          \ presentation."
+      - prompt: w@llet,A man in a light grey suit jacket layered over a textured white
+          knit sweater poses in a semi-formal look, holding a brown leather wallet
+          in his right hand. His left hand adjusts the lapel of his jacket, giving
+          the impression of placing the wallet into the inner pocket. The wallet is
+          angled forward to showcase its grain texture and stitched edges. A sleek
+          wristwatch peeks out on his left wrist, complementing the polished look.
+          Background is a neutral soft grey with diffused studio lighting, creating
+          a clean and professional aesthetic while keeping the focus on the wallet.
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 4
+      sample_steps: 25
+      num_frames: 1
+      fps: 1
+meta:
+  name: wallet-poses
+  version: '1.0'

optimizer.pt

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

wallet-poses.safetensors

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

wallet-poses_000000250.safetensors

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

wallet-poses_000000500.safetensors

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

wallet-poses_000000750.safetensors

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

wallet-poses_000001000.safetensors

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

wallet-poses_000001250.safetensors

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

wallet-poses_000001500.safetensors

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

wallet-poses_000001750.safetensors

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