job: extension
config:
  name: relax-piano
  process:
  - type: sd_trainer
    training_folder: output
    device: cuda:0
    network:
      type: lora
      linear: 32
      linear_alpha: 32
    save:
      dtype: float16
      save_every: 1000
      max_step_saves_to_keep: 5
      push_to_hub: true
    hf_repo_id: linhqyy/relax
    hf_private: false
    datasets:
    - folder_path: /workspace/Relax
      caption_ext: txt
      caption_dropout_rate: 0.05
      shuffle_tokens: false
      cache_latents_to_disk: true
      resolution:
      - 512
      - 768
      - 1024
    train:
      batch_size: 1
      steps: 5000
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0004
      ema_config:
        use_ema: true
        ema_decay: 0.99
      dtype: bf16
    model:
      name_or_path: black-forest-labs/FLUX.1-dev
      is_flux: true
      quantize: false
    sample:
      sampler: flowmatch
      sample_every: 500
      width: 1024
      height: 1024
      prompts:
      - This is a vibrant digital illustration featuring a serene forest scene. In
        the foreground, a tranquil pond reflects the surrounding lush greenery and
        a clear, blue sky with a few fluffy clouds. The pond is dotted with various-sized
        rocks and surrounded by vibrant flora, including orange and yellow flowers
        with large, glossy leaves. The water is calm, with gentle ripples and a few
        floating leaves. The central focus of the image is a cheerful, round, yellow
        sunfish-like creature with a smiling face, small pink flowers on its head,
        and closed eyes. It appears to be swimming contentedly in the pond, adding
        a whimsical and playful element to the scene. The background is a dense forest
        filled with tall, leafy trees with a mix of green and orange hues, suggesting
        autumn. The sunlight filters through the leaves, creating a dappled effect
        on the ground and the water. Small birds are seen flying or perched on the
        branches, adding to the peaceful ambiance. The overall style of the illustration
        is whimsical and cartoonish, with bright, bold colors and clean lines, emphasizing
        the tranquility and beauty of nature. The artist's attention to detail and
        the soft, almost dreamy lighting enhance the scene's charm.
      - This is a digital illustration depicting a cozy, nighttime study scene. At
        the center of the image is a young man with short, light blonde hair, wearing
        a white shirt and sitting in a simple wooden chair. He is focused on a laptop
        on a wooden desk, which is illuminated by a warm, yellow desk lamp. The room
        is dimly lit, suggesting it is late at night.  To the left of the desk, there
        is a potted plant with large, green leaves, adding a touch of nature to the
        otherwise urban environment. On the right side of the desk, there is another
        potted plant and a stack of books, indicating a love for reading.  Through
        the large window behind the man, a cityscape is visible, with tall skyscrapers
        illuminated by city lights, creating a contrast to the warm interior lighting.
        The sky outside is dark, with scattered raindrops visible, suggesting a rainy
        night. The overall ambiance of the image is serene and focused, with a blend
        of warm, cozy indoor lighting and the cool, urban exterior. The art style
        is detailed and realistic, with a focus on soft textures and subtle shading,
        creating a calming, inviting atmosphere.
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 20
meta:
  name: human_training
  version: '1.0'