job: extension
config:
  name: assa_model
  process:
  - type: sd_trainer
    training_folder: /content/drive/MyDrive/Flux
    device: cuda:0
    HST: photo
    network:
      type: lora
      linear: 16
      linear_alpha: 16
    save:
      dtype: float16
      save_every: 500
      max_step_saves_to_keep: 12
    datasets:
    - folder_path: /content/dataset
      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: 3000
      gradient_accumulation_steps: 1
      train_unet: true
      train_text_encoder: false
      content_or_style: balanced
      gradient_checkpointing: true
      noise_scheduler: flowmatch
      optimizer: adamw8bit
      lr: 0.0004
      skip_first_sample: true
      hf_repo_id: AlekseyCalvin/assa
      hf_private: true
      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: true
    sample:
      sampler: flowmatch
      sample_every: 250
      width: 1024
      height: 1024
      prompts:
      - kodachrome HST photo of woman Dark Kamala playing with time at the park
      - "HST style depiction of the musical sound of Little Oil, with centered title\
        \ text: \"Little Oil\u201D"
      - Raw kodachrome photo by Man Ray HST style, extremely photorealistic image
        of a hospital revolution, doctors break down smashing phones
      - HST style autochrome photo of thin confused ACS doctor saying word bubble
        text WHATS YOUR COST? at labyrinthine modern hospital hallway, dada color
        photo, twining hospital hallways, amateur photo
      - a dark murderous Koala HST style, building a log cabin out of music and teeth
        in the snow covered mountains of HST land Omsk
      - HST style kodachrome photo of Rowland S. Howard playing the guitar, on stage,
        singing a song, laser lights, punk rocker
      - HST style kodachrome color photo of Alexander Blok the poet holding a sign
        that says, 'truth is found in wine'
      - HST style autochrome vintage lifelike amateur film Photo of tiny drugged insectoid
        christ-like God inside of a wooden pillbox being fed ether poured by giant
        womankodachrome HST photo of woman Dark Kamala HST style caption text of "Historic
        Color Dev", analog kodachrome photo, blemished skin, amateur quality, cell
        phone photo
      - 'HST style depiction of the musical sound of Little Oil, disposable camera
        image, with centered title \Little Oil\ '
      - HST style photo of a cat
      neg: ''
      seed: 42
      walk_seed: true
      guidance_scale: 4
      sample_steps: 20
meta:
  name: assa_model
  version: '1.0'