Update config.yaml

config.yaml

@@ -1,70 +1,77 @@
-job: extension
-config:
-  name: bsh1
-  process:
-  - type: sd_trainer
-    training_folder: /root/ai-toolkit/modal_output
-    device: cuda:0
-    trigger_word: bsh1
-    network:
-      type: lora
-      linear: 32
-      linear_alpha: 32
-    save:
-      dtype: float16
-      save_every: 230
-      max_step_saves_to_keep: 12
-      push_to_hub: false
-    datasets:
-    - folder_path: /root/ai-toolkit/bsh1
-      caption_ext: txt
-      caption_dropout_rate: 0.1
-      shuffle_tokens: false
-      cache_latents_to_disk: true
-      resolution:
-      - 1024
-    train:
-      batch_size: 1
-      steps: 6670
-      gradient_accumulation: 1
-      train_unet: true
-      train_text_encoder: false
-      gradient_checkpointing: true
-      noise_scheduler: flowmatch
-      optimizer: adamw
-      timestep_type: shift
-      optimizer_params:
-        weight_decay: 0.0002
-      lr: 0.0002
-      skip_first_sample: true
-      disable_sampling: true
-      ema_config:
-        use_ema: true
-        ema_decay: 0.99
-      dtype: bf16
-    model:
-      name_or_path: /root/Qwen-Image
-      arch: qwen_image
-      quantize: false
-      quantize_te: false
-      low_vram: false
-    sample:
-      sampler: flowmatch
-      sample_every: 170
-      width: 1024
-      height: 1024
-      prompts:
-      - Photo of a young woman with long black hair, wearing a blue and white dress
-        with a white lace collar. She has a fair complexion and is looking directly
-        at the camera with a neutral expression. Her hair is adorned with small pink
-        flowers. The background is a gradient of blue to white, creating a soft and
-        dreamy atmosphere. The lighting is soft and even, highlighting her features.
-        The image has a high-quality, professional feel.
-      neg: ''
-      seed: 777
-      walk_seed: true
-      guidance_scale: 4
-      sample_steps: 30
-meta:
-  name: bsh1
-  version: '1.0'
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1.safetensors"
+filename = "bsh1.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000003910.safetensors"
+filename = "bsh1_000003910.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000004140.safetensors"
+filename = "bsh1_000004140.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000004370.safetensors"
+filename = "bsh1_000004370.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000004600.safetensors"
+filename = "bsh1_000004600.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000004830.safetensors"
+filename = "bsh1_000004830.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000005060.safetensors"
+filename = "bsh1_000005060.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000005290.safetensors"
+filename = "bsh1_000005290.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000005520.safetensors"
+filename = "bsh1_000005520.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000005750.safetensors"
+filename = "bsh1_000005750.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000005980.safetensors"
+filename = "bsh1_000005980.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000006210.safetensors"
+filename = "bsh1_000006210.safetensors"
+type = "loras"
+for = "qwen"
+
+[[models]]
+link = "https://huggingface.co/JealousType/bsh1/resolve/main/bsh1_000006440.safetensors"
+filename = "bsh1_000006440.safetensors"
+type = "loras"
+for = "qwen"