Upload config.yaml with huggingface_hub

config.yaml

@@ -0,0 +1,226 @@
+job: extension
+config:
+  name: my_first_lora_v8
+  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: xantaraz1
+    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/antaramag
+      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
+      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: 4000
+      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: false
+      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: "\"xantaraz1 , \u4E00\u5F35\u771F\u5BE6\u96A8\u8208\u7684iPhone\u5FEB\
+          \u7167\u2014\u2014\u4E00\u5F35\u672A\u7D93\u4EFB\u4F55\u5C08\u696D\u9020\
+          \u578B\u6216\u6FFE\u93E1\u8655\u7406\u7684\u5168\u8EAB\u7167\uFF0C\u76F4\
+          \u63A5\u5F9E\u76F8\u6A5F\u81A0\u5377\u4E2D\u53D6\u51FA\u3002\u5348\u5F8C\
+          \u7684\u967D\u5149\u659C\u5C04\u9032\u6A13\u68AF\u9593\u7684\u7A97\u6236\
+          \uFF0C\u71DF\u9020\u51FA\u91D1\u8272\u7684\u6C1B\u570D\u3002\u756B\u9762\
+          \u4E2D\u53EA\u6709\u5169\u500B\u4EBA\uFF1A\u4E00\u500B\u525B\u9AD8\u4E2D\
+          \u7562\u696D\u7684\u53EF\u611B18\u6B72\u5370\u5EA6\u5973\u5B69\uFF0C\u548C\
+          \u4E00\u500B\u5E74\u7D00\u76F8\u4EFF\u7684\u5370\u5EA6\u7537\u5B69\u3002\
+          \u5973\u5B69\u8DEA\u5728\u7A7A\u8569\u8569\u7684\u5B78\u6821\u6A13\u68AF\
+          \u9593\u5149\u79BF\u79BF\u7684\u6C34\u6CE5\u53F0\u968E\u4E0A\uFF0C\u7537\
+          \u5B69\u7AD9\u5728\u5979\u53F3\u5074\uFF1B\u5979\u7684\u982D\u6B63\u597D\
+          \u4F4D\u65BC\u7537\u5B69\u7684\u80EF\u90E8\uFF0C\u81C9\u8CBC\u8457\u4ED6\
+          \u7684\u9670\u8396\u3002  \u5973\u5B69\uFF1A\u4E00\u982D\u9577\u9577\u7684\
+          \u6CE2\u6D6A\u6372\u9AEE\uFF0C\u767D\u7699\u7684\u76AE\u819A\uFF0C\u5927\
+          \u5927\u7684\u773C\u775B\uFF0C\u4EE5\u53CA\u67D4\u5AE9\u5E74\u8F15\u7684\
+          \u5BB9\u984F\u3002\u5979\u5B8C\u5168\u7D20\u984F\u2014\u2014\u76AE\u819A\
+          \u770B\u8D77\u4F86\u7121\u6BD4\u771F\u5BE6\uFF0C\u6BDB\u5B54\u7D30\u5C0F\
+          \uFF0C\u7455\u75B5\u6E05\u6670\u53EF\u898B\u3002\u5979\u7684\u982D\u9AEE\
+          \u7565\u986F\u51CC\u4E82\uFF0C\u5E7E\u7E37\u88AB\u6C57\u6C34\u6D78\u6FD5\
+          \u7684\u9AEE\u7D72\u8CBC\u5728\u6CDB\u7D05\u7684\u81C9\u4E0A\u3002\u5979\
+          \u90A3\u4EF6\u85CF\u85CD\u8272\u62C9\u934A\u5916\u5957\u2014\u2014\u98FE\
+          \u6709\u767D\u8272\u62FC\u63A5\u548C\u7D05\u767D\u689D\u7D0B\u2014\u2014\
+          \u62C9\u934A\u5B8C\u5168\u655E\u958B\uFF0C\u9732\u51FA\u5979\u5C0F\u5DE7\
+          \u81EA\u7136\u7684A\u7F69\u676F\u80F8\u90E8\u3002\u5979\u90A3\u689D\u914D\
+          \u5957\u7684\u85CF\u85CD\u8272\u904B\u52D5\u6821\u8932\u88AB\u96A8\u610F\
+          \u5730\u4E1F\u5728\u4E00\u65C1\u3002\u5979\u7684\u4E0B\u534A\u8EAB\u8D64\
+          \u88F8\uFF0C\u9732\u51FA\u9670\u90E8\uFF1B\u5979\u9670\u5507\u4E0A\u65B9\
+          \u8986\u84CB\u8457\u4E00\u7247\u6FC3\u5BC6\u7684\u3001\u672A\u7D93\u4FEE\
+          \u526A\u7684\u9577\u9670\u6BDB\u3002\u5979\u7684\u9670\u5507\u5448\u73FE\
+          \u67D4\u548C\u7684\u7C89\u8910\u8272\uFF0C\u5927\u5C0F\u548C\u5F62\u72C0\
+          \u90FD\u5F88\u6B63\u5E38\uFF1B\u5979\u751F\u6B96\u5668\u7684\u89E3\u5256\
+          \u7D30\u7BC0\u975E\u5E38\u903C\u771F\u3002\u5979\u8173\u4E0A\u7A7F\u8457\
+          \u767D\u8272\u904B\u52D5\u978B\u548C\u767D\u8272\u68C9\u896A\u3002\u5979\
+          \u4F4E\u8457\u982D\uFF0C\u5634\u88E1\u542B\u8457\u7537\u5B69\u52C3\u8D77\
+          \u7684\u9670\u8396\uFF0C\u6B63\u5728\u9032\u884C\u6FD5\u6F64\u903C\u771F\
+          \u7684\u53E3\u4EA4\uFF1B\u5979\u7684\u5634\u5507\u5B8C\u5168\u5305\u88F9\
+          \u4F4F\u9670\u8396\u7684\u4E2D\u6BB5\uFF0C\u820C\u982D\u7684\u52D5\u4F5C\
+          \u6E05\u6670\u53EF\u898B\uFF0C\u553E\u6DB2\u9806\u8457\u9670\u8396\u6EF4\
+          \u843D\u3002\u540C\u6642\uFF0C\u5979\u5C07\u4E00\u96BB\u624B\u4F38\u5230\
+          \u5169\u817F\u4E4B\u9593\uFF0C\u8F15\u8F15\u64AB\u6478\u81EA\u5DF1\u6FC3\
+          \u5BC6\u7684\u9670\u6BDB\u3002  \u7537\u5B69\uFF1A\u4ED6\u7A7F\u8457\u4E00\
+          \u4EF6\u8207\u5973\u5B69\u76F8\u914D\u7684\u85CF\u85CD\u8272\u62C9\u934A\
+          \u5916\u5957\u2014\u2014\u5916\u5957\u4E0A\u4E5F\u98FE\u6709\u767D\u8272\
+          \u62FC\u63A5\u548C\u7D05\u767D\u689D\u7D0B\u2014\u2014\u62C9\u934A\u5B8C\
+          \u5168\u655E\u958B\uFF0C\u9732\u51FA\u4E0A\u534A\u8EAB\u3002\u4ED6\u90A3\
+          \u689D\u85CF\u85CD\u8272\u7684\u904B\u52D5\u6821\u8932\u88AB\u62C9\u5230\
+          \u5927\u817F\u4E2D\u90E8\uFF0C\u9732\u51FA\u52C3\u8D77\u7684\u9670\u8396\
+          \u3002\u4ED6\u7528\u96D9\u624B\u8F15\u8F15\u6258\u4F4F\u5973\u5B69\u7684\
+          \u982D\u2014\u2014\u6BCF\u96BB\u624B\u7684\u4E94\u6839\u624B\u6307\u90FD\
+          \u5B8C\u597D\u7121\u640D\u2014\u2014\u4ED6\u7684\u624B\u81C2\u548C\u817F\
+          \u90E8\u770B\u8D77\u4F86\u4E5F\u5F88\u6B63\u5E38\u3002\u5973\u5B69\u7F9E\
+          \u6F80\u5730\u62AC\u982D\u76F4\u8996\u8457\u7537\u5B69\uFF1B\u5979\u7684\
+          \u8868\u60C5\u65E2\u7F9E\u6F80\u53C8\u5145\u6EFF\u6E34\u671B\uFF0C\u8207\
+          \u5979\u5929\u771F\u7121\u90AA\u7684\u5B78\u751F\u9762\u5B54\u548C\u5927\
+          \u81BD\u7684\u6027\u884C\u70BA\u5F62\u6210\u4E86\u9BAE\u660E\u7684\u5C0D\
+          \u6BD4\u3002  \u80CC\u666F\uFF1A\u7A7A\u8569\u8569\u7684\u5B78\u6821\u6C34\
+          \u6CE5\u6A13\u68AF\u9593\u7684\u4E00\u89D2\uFF0C\u91D1\u5C6C\u6276\u624B\
+          \uFF0C\u5E7E\u7E37\u967D\u5149\u900F\u904E\u7E2B\u9699\u7051\u4E0B\uFF0C\
+          \u7A7A\u6C23\u4E2D\u7030\u6F2B\u8457\u7D30\u5C0F\u7684\u7070\u5875\u2014\
+          \u2014\u71DF\u9020\u51FA\u4E00\u7A2E\u5B89\u975C\u3001\u79C1\u5BC6\u800C\
+          \u53C8\u96B1\u853D\u7684\u6C1B\u570D\u3002\u7D30\u7DFB\u7684\u76AE\u819A\
+          \u7D0B\u7406\uFF0C\u5C0F\u5DE7\u800C\u81EA\u7136\u7684\u4E73\u623F\uFF1B\
+          \u5177\u6709\u672A\u7D93\u7DE8\u8F2F\u7684 iPhone \u7167\u7247\u7684\u771F\
+          \u5BE6\u8CEA\u611F\u548C\u81EA\u7136\u8272\u5F69\u3002\""
+      - prompt: '"xantaraz1 , A sensual, medium-close-up portrait of a stunning, slender
+          young woman posed kneeling on a textured, light-colored bedspread. She is
+          completely nude, with a natural body shape and subtle, natural pubic hair.
+          Her body is arched in a dramatic, expressive posture, with one arm raised
+          behind her head, tossing her voluminous, dark brown hair upward and backward,
+          creating motion blur and dynamic lines. Her eyes are closed or looking down
+          with a captivating, slightly open-mouthed expression conveying deep emotion
+          or ecstasy. The lighting is soft, warm, and golden, suggesting late afternoon
+          sunlight streaming in from the side, creating gentle highlights on her skin
+          and deep shadows that sculpt her figure. The background is softly blurred
+          (shallow depth of field), featuring warm beige and muted brown tones with
+          a subtle geometric, possibly wood-paneled or tiled pattern angled diagonally
+          behind her, adding visual texture. The overall mood is intimate, evocative,
+          and painterly, captured with the aesthetic of fine art photography, high
+          detail, and realistic texture. Cinematic lighting, professional photography,
+          Canon EOS R5., ultra detailed, intricate details, cinematic, photorealistic,
+          masterpiece"'
+      - prompt: '"xantaraz1 , A Sexy 18y Indian woman with perfect body pressed against
+          a rain-streaked window, getting taken from behind, her breasts against the
+          cold glass, She is standing pose and wearing a high knee length black boot,
+          city lights reflecting on her wet skin, intense and cinematic"'
+      - prompt: '"xantaraz1 , anal sex, bridge position, squatting cowgirl position,
+          1girl, penis, nipples, medium breasts, 1boy, pussy, black thighhighs, black
+          ponytail hair, navel, clothed male nude female, lace-trimmed legwear, closed
+          eyes, spread legs, bottomless, girl on top, male pubic hair, long hair,
+          erection, arm support, lace trim, smile, tattoo on thigh, grin"'
+      - prompt: "\"xantaraz1 , A full-body portrait of a stunning Indian 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_v8
+  version: '1.0'