Add terry.yaml configuration file

terry.yaml

@@ -0,0 +1,75 @@
+---
+job: extension
+config:
+  name: "terry"
+  process:
+    - type: 'sd_trainer'
+      training_folder: "output"
+      device: cuda:0
+      network:
+        type: "lora"
+        linear: 16
+        linear_alpha: 16
+      save:
+        dtype: float16 
+        save_every: 500 
+        max_step_saves_to_keep: 4 
+        push_to_hub: true 
+        hf_repo_id: maver1chh/terry
+        hf_private: false  
+      datasets:
+        - folder_path: "/home/user0/PDam/ai-toolkit/data/terry"
+          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: 6000 
+        gradient_accumulation_steps: 1
+        train_unet: true
+        train_text_encoder: false  # probably won't work with flux
+        gradient_checkpointing: true  # need the on unless you have a ton of vram
+        noise_scheduler: "flowmatch" 
+        optimizer: "adamw8bit"
+        lr: 1e-4
+        # uncomment this to skip the pre training sample
+#        skip_first_sample: true
+        # uncomment to completely disable sampling
+#        disable_sampling: true
+        # uncomment to use new vell curved weighting. Experimental but may produce better results
+#        linear_timesteps: true
+
+        # ema will smooth out learning, but could slow it down. Recommended to leave on.
+        ema_config:
+          use_ema: true
+          ema_decay: 0.99
+
+        # will probably need this if gpu supports it for flux, other dtypes may not work correctly
+        dtype: bf16
+      model:
+        # huggingface model name or path
+        name_or_path: "black-forest-labs/FLUX.1-dev"
+        is_flux: true
+        quantize: true  # run 8bit mixed precision
+#        low_vram: true  # uncomment this if the GPU is connected to your monitors. It will use less vram to quantize, but is slower.
+      sample:
+        sampler: "flowmatch" # must match train.noise_scheduler
+        sample_every: 250 # sample every this many steps
+        width: 1024
+        height: 1024
+        prompts:
+          # you can add [trigger] to the prompts here and it will be replaced with the trigger word
+#          - "[trigger] holding a sign that says 'I LOVE PROMPTS!'"\
+          - "Spider-Man as an astronaut traveling to the universe, wearing a futuristic white and red space suit with the signature spider logo on his chest. The suit is highly detailed, with glowing blue lights and web-like textures across the helmet visor and armor plates. Spider-Man is floating weightlessly in deep space, surrounded by vibrant nebulas, distant stars, and glowing planets in the background. His pose is dynamic and heroic, one hand reaching out as if he’s about to shoot a web in zero gravity. The helmet reflects the cosmic scenery, and faint spider-web patterns glimmer in the cosmic light. The atmosphere is awe-inspiring, cinematic, and full of sci-fi realism, blending Spider-Man's iconic superhero aesthetics with the wonders of space."
+          - "A superhero sitting at a wooden table in a cozy sushi restaurant, eating sushi with chopsticks. The superhero wears a vibrant, detailed costume with a cape and mask, reflecting their iconic persona. The scene is illustrated in the style of Vincent van Gogh’s paintings, featuring swirling brush strokes, rich colors, and textured patterns reminiscent of Starry Night or Café Terrace at Night. The sushi rolls are meticulously painted with vibrant reds, greens, and whites, arranged beautifully on a wooden platter. The background glows warmly, with the restaurant interior full of swirling lights, lanterns, and dreamy textures. The superhero's pose is relaxed and joyful, appreciating the sushi in this unique, painterly world. The overall atmosphere is whimsical, colorful, and artistic."
+        neg: ""  # not used on flux
+        seed: 42
+        walk_seed: true
+        guidance_scale: 4
+        sample_steps: 20
+# you can add any additional meta info here. [name] is replaced with config name at top
+meta:
+  name: "[name]"
+  version: '1.0'