Update config.yaml

config.yaml

@@ -1,87 +1,6 @@
-job: extension
-config:
-  name: flux_dev_finetune
-  process:
-  - type: sd_trainer
-    training_folder: /content/output
-    performance_log_every: 50
-    device: cuda:0
-    network:
-      type: lora
-      linear: 16
-      linear_alpha: 32
-    save:
-      dtype: float16
-      save_every: 250
-      max_step_saves_to_keep: 4
-    datasets:
-    - folder_path: /content/dataset
-      caption_ext: txt
-      caption_dropout_rate: 0.05
-      shuffle_tokens: true
-      cache_latents_to_disk: true
-      resolution:
-      - 512
-    train:
-      batch_size: 8
-      steps: 1000
-      gradient_accumulation_steps: 2
-      train_unet: true
-      train_text_encoder: false
-      content_or_style: balanced
-      gradient_checkpointing: true
-      noise_scheduler: flowmatch
-      optimizer: adamw
-      lr: 0.0001
-      ema_config:
-        use_ema: true
-        ema_decay: 0.99
-      dtype: float16
-    model:
-      name_or_path: black-forest-labs/FLUX.1-dev
-      is_flux: true
-      quantize: true
-    sample:
-      sampler: flowmatch
-      sample_every: 500
-      width: 512
-      height: 512
-      prompts:
-      - An intricate floral motif inspired by traditional Mughal art, with ornate
-        paisleys and delicate vines in shades of deep maroon, emerald green, and gold
-        on a cream background, reminiscent of classic shalwar kameez patterns.
-      - A luxurious pattern featuring block-printed motifs of lotus flowers and mandalas
-        in muted pastel tones of pink, peach, and mint green, evoking the rich textile
-        traditions of Pakistans artisanal heritage.
-      - A sophisticated fusion of contemporary minimalism with traditional Pakistani
-        embroidery, featuring delicate floral butis scattered over a soft linen base
-        in pastel tones, perfect for summer tunics.
-      - A sophisticated fusion of contemporary minimalism with traditional Pakistani
-        embroidery, featuring delicate floral butis scattered over a soft linen base
-        in pastel tones, perfect for summer tunics.
-      - A Balochi-style geometric pattern with diamond and triangle shapes in earthy
-        tones of rust, brown, and cream, paying homage to the nomadic craftsmanship
-        of Pakistans western regions
-      - A playful polka dot pattern with oversized, irregular dots in bold primary
-        colors like red, blue, and yellow, set against a crisp white background, giving
-        a retro yet fresh vibe for summer dresses
-      - A sophisticated tartan pattern in deep shades of forest green, navy, and burgundy,
-        perfect for stylish woolen skirts and scarves, channeling a refined Scottish
-        heritage look.
-      - A Kashmiri shawl-inspired pattern featuring delicate swirls of vines and paisleys
-        in soft pastel shades, woven into an intricate border design for a light,
-        airy, and elegant fabric.
-      - a serene mountain landscape pattern with stylized pine trees, snow-capped
-        peaks, and tranquil lakes, in earthy tones and soft pastels, creating a calming
-        atmosphere
-      - A hand-painted style floral pattern inspired by the phulkari art of Punjab,
-        with large blooming flowers in bright shades of fuchsia, orange, and yellow,
-        set against a neutral beige background.
-      neg: ''
-      seed: 42
-      walk_seed: true
-      guidance_scale: 4
-      sample_steps: 20
-meta:
-  name: flux_dev_finetune
-  version: '1.0'
+model_type: "diffusion"
+pipeline_tag: "text-to-image"
+vae:
+  type: "AutoencoderKL"
+scheduler:
+  type: "FlowMatchEulerDiscreteScheduler"