Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ 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
+loss_log.db filter=lfs diff=lfs merge=lfs -text

.job_config.json

@@ -0,0 +1,143 @@
+{
+  "job": "extension",
+  "config": {
+    "name": "rupsaz1_lora_v1",
+    "process": [
+      {
+        "type": "diffusion_trainer",
+        "training_folder": "/app/ai-toolkit/output",
+        "sqlite_db_path": "/app/ai-toolkit/aitk_db.db",
+        "device": "cuda",
+        "trigger_word": "rupsaz1",
+        "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": 4,
+          "save_format": "diffusers",
+          "push_to_hub": false
+        },
+        "datasets": [
+          {
+            "folder_path": "/app/ai-toolkit/datasets/rupss",
+            "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
+            ],
+            "controls": [],
+            "shrink_video_to_frames": true,
+            "num_frames": 1,
+            "flip_x": false,
+            "flip_y": false,
+            "num_repeats": 1
+          }
+        ],
+        "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": "weighted",
+          "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": false,
+            "ema_decay": 0.99
+          },
+          "skip_first_sample": false,
+          "force_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,
+          "loss_type": "mse"
+        },
+        "logging": {
+          "log_every": 1,
+          "use_ui_logger": true
+        },
+        "model": {
+          "name_or_path": "Tongyi-MAI/Z-Image-Turbo",
+          "quantize": true,
+          "qtype": "qfloat8",
+          "quantize_te": true,
+          "qtype_te": "qfloat8",
+          "arch": "zimage:turbo",
+          "low_vram": true,
+          "model_kwargs": {},
+          "layer_offloading": false,
+          "layer_offloading_text_encoder_percent": 1,
+          "layer_offloading_transformer_percent": 1,
+          "assistant_lora_path": "ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors"
+        },
+        "sample": {
+          "sampler": "flowmatch",
+          "sample_every": 250,
+          "width": 1024,
+          "height": 1024,
+          "samples": [
+            {
+              "prompt": "Photo of a nude Indian woman 21y old standing in a bedroom, facing the camera, medium wide shot, full body, black hair, dark eyes, medium breasts, slender physique, medium skin tone, brown eyes, dark eyebrows, small areola, standing with her legs together, hands at her sides, looking at the camera, natural lighting, bright and airy room, white walls, large window with sheer white curtains, white bed with a wooden frame, white bed linens, wooden floor, potted plant in the background, casual and relaxed pose, confident expression, soft shadows, high quality, jpeg artifacts, no watermark"
+            },
+            {
+              "prompt": "Photo of a Indian 21y woman on her knees in a bathroom, giving a blowjob to a man standing in front of her. The woman has long black hair and a slender physique, with medium-sized breasts. The man is naked, with a large erect penis. The woman is performing oral sex on the man, with her eyes closed and her lips wrapped around his penis. The bathroom has a modern design, a wooden floor. The lighting is soft and warm, with a gentle glow on the woman's skin. The image is high quality and has a watermark in the bottom right corner."
+            },
+            {
+              "prompt": "A young woman with an oval-shaped face, soft rounded cheeks, almond-shaped eyes, and naturally full eyebrows, wearing a calm, confident expression. She has dark brown to black hair with a natural texture, worn loose and flowing past the shoulders. Her skin tone is medium warm brown with natural, even variation and a realistic finish. She has an average, proportionate body frame with a gently toned torso and smooth contours. She is posing indoors with both arms raised and hands resting behind her head, chest lifted and posture relaxed yet confident. She is dressed in a black lace lingerie set with delicate floral embroidery, including a structured lace bra and matching lace briefs. Soft natural window light casts gentle shadows across her body, creating a warm, cinematic mood. The setting is a minimal bedroom interior with sheer curtains, shallow depth of field, photorealistic style, high detail, natural skin texture, realistic lighting, 85mm lens look, ultra-high resolution."
+            },
+            {
+              "prompt": "Photo of a nude woman with long black hair kneeling on a bathroom floor, facing the camera, looking directly at the camera with a neutral expression. She has a medium skin tone and a slender, toned physique. Her breasts are medium-sized with small areolas. The bathroom has gray tiled walls and a white toilet in the background. The lighting is soft and even, creating a calm and intimate atmosphere. The woman's pose is natural and confident. The image has a watermark in the bottom right corner that reads \"www.Femjoy.com\"."
+            },
+            {
+              "prompt": "Ultra-realistic close-up portrait of a young Indian woman with an oval face, soft rounded cheeks, almond-shaped dark brown eyes, naturally full eyebrows, and a straight, proportionate nose, subtle natural lips with gentle definition. Medium warm brown Indian skin tone with realistic texture, visible pores, soft highlights, and natural color variation. Dark brown to black hair with a natural texture, softly framing the face. Minimal makeup, natural glow, calm confident expression, direct eye contact with the camera. Soft diffused window light illuminating the face, cinematic shadows, shallow depth of field, photorealistic skin detail, 85mm lens look, ultra-high resolution, studio-quality realism."
+            }
+          ],
+          "neg": "",
+          "seed": 42,
+          "walk_seed": true,
+          "guidance_scale": 1,
+          "sample_steps": 8,
+          "num_frames": 1,
+          "fps": 1
+        }
+      }
+    ]
+  },
+  "meta": {
+    "name": "[name]",
+    "version": "1.0"
+  }
+}

config.yaml

@@ -0,0 +1,155 @@
+job: extension
+config:
+  name: rupsaz1_lora_v1
+  process:
+  - type: diffusion_trainer
+    training_folder: /app/ai-toolkit/output
+    sqlite_db_path: /app/ai-toolkit/aitk_db.db
+    device: cuda
+    trigger_word: rupsaz1
+    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: 4
+      save_format: diffusers
+      push_to_hub: false
+    datasets:
+    - folder_path: /app/ai-toolkit/datasets/rupss
+      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
+      controls: []
+      shrink_video_to_frames: true
+      num_frames: 1
+      flip_x: false
+      flip_y: false
+      num_repeats: 1
+    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: weighted
+      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: false
+        ema_decay: 0.99
+      skip_first_sample: false
+      force_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
+      loss_type: mse
+    logging:
+      log_every: 1
+      use_ui_logger: true
+    model:
+      name_or_path: Tongyi-MAI/Z-Image-Turbo
+      quantize: true
+      qtype: qfloat8
+      quantize_te: true
+      qtype_te: qfloat8
+      arch: zimage:turbo
+      low_vram: true
+      model_kwargs: {}
+      layer_offloading: false
+      layer_offloading_text_encoder_percent: 1
+      layer_offloading_transformer_percent: 1
+      assistant_lora_path: ostris/zimage_turbo_training_adapter/zimage_turbo_training_adapter_v2.safetensors
+    sample:
+      sampler: flowmatch
+      sample_every: 250
+      width: 1024
+      height: 1024
+      samples:
+      - prompt: Photo of a nude Indian woman 21y old standing in a bedroom, facing
+          the camera, medium wide shot, full body, black hair, dark eyes, medium breasts,
+          slender physique, medium skin tone, brown eyes, dark eyebrows, small areola,
+          standing with her legs together, hands at her sides, looking at the camera,
+          natural lighting, bright and airy room, white walls, large window with sheer
+          white curtains, white bed with a wooden frame, white bed linens, wooden
+          floor, potted plant in the background, casual and relaxed pose, confident
+          expression, soft shadows, high quality, jpeg artifacts, no watermark
+      - prompt: Photo of a Indian 21y woman on her knees in a bathroom, giving a blowjob
+          to a man standing in front of her. The woman has long black hair and a slender
+          physique, with medium-sized breasts. The man is naked, with a large erect
+          penis. The woman is performing oral sex on the man, with her eyes closed
+          and her lips wrapped around his penis. The bathroom has a modern design,
+          a wooden floor. The lighting is soft and warm, with a gentle glow on the
+          woman's skin. The image is high quality and has a watermark in the bottom
+          right corner.
+      - prompt: A young woman with an oval-shaped face, soft rounded cheeks, almond-shaped
+          eyes, and naturally full eyebrows, wearing a calm, confident expression.
+          She has dark brown to black hair with a natural texture, worn loose and
+          flowing past the shoulders. Her skin tone is medium warm brown with natural,
+          even variation and a realistic finish. She has an average, proportionate
+          body frame with a gently toned torso and smooth contours. She is posing
+          indoors with both arms raised and hands resting behind her head, chest lifted
+          and posture relaxed yet confident. She is dressed in a black lace lingerie
+          set with delicate floral embroidery, including a structured lace bra and
+          matching lace briefs. Soft natural window light casts gentle shadows across
+          her body, creating a warm, cinematic mood. The setting is a minimal bedroom
+          interior with sheer curtains, shallow depth of field, photorealistic style,
+          high detail, natural skin texture, realistic lighting, 85mm lens look, ultra-high
+          resolution.
+      - prompt: Photo of a nude woman with long black hair kneeling on a bathroom
+          floor, facing the camera, looking directly at the camera with a neutral
+          expression. She has a medium skin tone and a slender, toned physique. Her
+          breasts are medium-sized with small areolas. The bathroom has gray tiled
+          walls and a white toilet in the background. The lighting is soft and even,
+          creating a calm and intimate atmosphere. The woman's pose is natural and
+          confident. The image has a watermark in the bottom right corner that reads
+          "www.Femjoy.com".
+      - prompt: Ultra-realistic close-up portrait of a young Indian woman with an
+          oval face, soft rounded cheeks, almond-shaped dark brown eyes, naturally
+          full eyebrows, and a straight, proportionate nose, subtle natural lips with
+          gentle definition. Medium warm brown Indian skin tone with realistic texture,
+          visible pores, soft highlights, and natural color variation. Dark brown
+          to black hair with a natural texture, softly framing the face. Minimal makeup,
+          natural glow, calm confident expression, direct eye contact with the camera.
+          Soft diffused window light illuminating the face, cinematic shadows, shallow
+          depth of field, photorealistic skin detail, 85mm lens look, ultra-high resolution,
+          studio-quality realism.
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 1
+      sample_steps: 8
+      num_frames: 1
+      fps: 1
+meta:
+  name: rupsaz1_lora_v1
+  version: '1.0'

loss_log.db

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

optimizer.pt

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

pid.txt

@@ -0,0 +1 @@
+1318

rupsaz1_lora_v1.safetensors

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

rupsaz1_lora_v1_000001000.safetensors

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

rupsaz1_lora_v1_000001250.safetensors

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

rupsaz1_lora_v1_000001500.safetensors

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

rupsaz1_lora_v1_000001750.safetensors

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