Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,160 @@
+job: extension
+config:
+  name: my_first_lora_v1
+  process:
+  - type: diffusion_trainer
+    training_folder: /teamspace/studios/this_studio/ai-toolkit/output
+    sqlite_db_path: /teamspace/studios/this_studio/ai-toolkit/aitk_db.db
+    device: cuda
+    trigger_word: srutixzit1
+    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: /teamspace/studios/this_studio/ai-toolkit/datasets/sruti
+      mask_path: null
+      mask_min_value: 0.1
+      default_caption: ''
+      caption_ext: txt
+      caption_dropout_rate: 0.03
+      cache_latents_to_disk: false
+      is_reg: false
+      network_weight: 1
+      resolution:
+      - 1024
+      - 768
+      - 512
+      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: 3000
+      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: ultra realistic portrait of a beautiful Indian 20y woman, soft natural
+          skin texture, detailed face, expressive eyes, subtle makeup, cinematic lighting,
+          85mm lens, shallow depth of field, high detail, photorealistic, natural
+          skin pores
+      - prompt: full body shot of a young Indian 20y nude woman standing naturally,
+          perfect body proportions, realistic anatomy, long legs, natural pose, detailed
+          hands and fingers, soft lighting, high resolution, photorealistic, balanced
+          composition
+      - prompt: "A young Indian woman, elegant and natural, is on her knees beside\
+          \ a fully naked man beneath a warm bathroom shower. They are completely\
+          \ nude in a private, intimate moment. The scene takes place inside a modern\
+          \ yet warm bathroom at dawn \u2014 soft morning light streams through a\
+          \ large window or diffuses from the shower curtain. She performs a deep\
+          \ throat swallow on his large penis with focused intensity and quiet passion\
+          \ in her eyes. Her hair is wet, glossy, and cascading over her shoulders\
+          \ and back, clinging to her neck and face, catching the water droplets and\
+          \ soft light. The atmosphere is intimate, sensual, and peaceful \u2014 realistic\
+          \ skin texture, steam rising from the shower, visible moisture on her skin,\
+          \ natural lighting, subtle details of hands, expression, and body language.\
+          \ Style: photorealistic, high detail, cinematic soft glow, warm tones, atmospheric\
+          \ humidity."
+      - prompt: a confident Indian 20y nude woman in a relaxed indoor setting, no
+          outfit, natural body posture, soft window light, realistic skin tones, detailed
+          anatomy, elegant and sensual mood, cinematic shadows, ultra realistic
+      - prompt: "A full-body portrait of a stunning Indian 20y woman with long, flowing\
+          \ black hair cascading down her back, standing nude and poised in a serene,\
+          \ softly lit bedroom. Her body is sculpted with perfect, natural curves\
+          \ \u2014 full, round breasts with dark brownish areolas and nipples, a smooth\
+          \ midriff leading to a gently curved waist, and a deeply toned, naturally\
+          \ darkened pubic area that accentuates her feminine form. Her legs are long\
+          \ and slender, ending in a firm, rounded ass that subtly tenses as she stands\
+          \ with her arms relaxed at her sides, fingers slightly curled, exuding quiet\
+          \ confidence and sensual grace. Her gaze is direct and unflinching, meeting\
+          \ the viewer\u2019s eyes with a calm, alluring intensity that draws you\
+          \ in.  The environment is minimalist and intimate, set against a soft, neutral\
+          \ beige background that enhances the natural tones of her skin and hair.\
+          \ Behind her, sheer white curtains filter in diffused daylight, casting\
+          \ gentle shadows across the room, while a simple bed with rumpled white\
+          \ linens and a potted green plant in the corner add subtle domestic warmth\
+          \ without distracting from the central focus. The lighting is soft and even,\
+          \ highlighting the smooth texture of her skin, the subtle contours of her\
+          \ anatomy, and the natural sheen of her hair \u2014 all rendered with photorealistic\
+          \ precision to emphasize her flawless, voluptuous form.  This is a portrait\
+          \ of raw, unfiltered beauty \u2014 a celebration of the female body in its\
+          \ most natural state. Her posture is relaxed yet commanding, her expression\
+          \ serene and self-assured, embodying both innocence and sensuality. The\
+          \ composition is vertically framed, drawing attention to her full, symmetrical\
+          \ silhouette, while the muted, monochromatic backdrop ensures every curve,\
+          \ every detail of her anatomy \u2014 from the delicate arch of her collarbone\
+          \ to the subtle definition of her hips \u2014 is rendered with exquisite\
+          \ clarity. She stands as a living sculpture, a vision of erotic elegance,\
+          \ inviting the viewer to appreciate the harmony of her form and the quiet\
+          \ power of her presence.\""
+      neg: ''
+      seed: 42
+      walk_seed: true
+      guidance_scale: 1
+      sample_steps: 8
+      num_frames: 1
+      fps: 1
+meta:
+  name: my_first_lora_v1
+  version: '1.0'