Upload training config

training_config.yaml

@@ -0,0 +1,90 @@
+---
+job: extension
+config:
+  # this name will be the folder and filename name
+  name: "nanase"
+  process:
+    - type: 'sd_trainer'
+      # root folder to save training sessions/samples/weights
+      training_folder: "/opt/stationthis/jobs/training-1774649168419/output"
+      # uncomment to see performance stats in the terminal every N steps
+#      performance_log_every: 1000
+      device: cuda:0
+      # if a trigger word is specified, it will be added to captions of training data if it does not already exist
+      # alternatively, in your captions you can add [trigger] and it will be replaced with the trigger word
+      trigger_word: "nanase"
+      network:
+        type: "lora"
+        linear: 32
+        linear_alpha: 32
+      save:
+        dtype: float16 # precision to save
+        save_every: 250 # save every this many steps (min of 250 or total steps)
+        max_step_saves_to_keep: 2 # how many intermittent saves to keep
+        push_to_hub: false #change this to True to push your trained model to Hugging Face.
+        # You can either set up a HF_TOKEN env variable or you'll be prompted to log-in         
+#       hf_repo_id: your-username/your-model-slug
+#       hf_private: true #whether the repo is private or public
+      datasets:
+        # datasets are a folder of images. captions need to be txt files with the same name as the image
+        # for instance image2.jpg and image2.txt. Only jpg, jpeg, and png are supported currently
+        # images will automatically be resized and bucketed into the resolution specified
+        # on windows, escape back slashes with another backslash so
+        # "C:\\path\\to\\images\\folder"
+        - folder_path: "/opt/stationthis/jobs/training-1774649168419/dataset"
+          caption_ext: "txt"
+          caption_dropout_rate: 0.05  # will drop out the caption 5% of time
+          shuffle_tokens: false  # shuffle caption order, split by commas
+          cache_latents_to_disk: true  # leave this true unless you know what you're doing
+          resolution: [ 512, 768, 1024 ]  # flux enjoys multiple resolutions
+      train:
+        batch_size: 1
+        steps: 4000  # total number of steps to train 500 - 4000 is a good range
+        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" # for training only
+        optimizer: "adamw8bit"
+        lr: 1e-4
+        # Skip baseline samples, only sample during/after training
+        skip_first_sample: true
+        # set to true 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: 4000 # sample at final step (TRAIN_STEPS - 1 for 0-indexed)
+        width: 1024
+        height: 1024
+        prompts:
+          # Dynamic prompts from dataset captions (injected by launch-training.js)
+          # These will be replaced with actual captions for HuggingFace samples
+          - "nanase, This is a colored pencil drawing of a young girl with light brown, shoulder-length hair, smiling widely with her eyes closed. She has fair skin and is depicted in a school uniform, wearing a light blue shirt with a blue and white striped tie. Her left hand is raised to her head, adjusting her hair. The background features soft, green and yellow brushstrokes, adding a sense of warmth and happiness to the image. The girl's expression is joyful and carefree. The artist's signature, \"YUMI 2020,\" is in the bottom left corner. The overall style is cute and anime-inspired."
+          - "nanase, This is a digital watercolor-style anime drawing of a young woman with long, dark blue hair tied in a high ponytail. She has fair skin, a slightly blushed cheek, and is gazing to the right with a calm expression. She wears a light gray sailor-style school uniform with a high collar and a pink cherry blossom pin on the left side. The background features abstract, light pink and gray patterns resembling tree branches. The artist's signature, \"LILY KOMACHI,\" is written in black on the left side. The overall style is soft and delicate, with gentle color gradients and fine line work."
+          - "nanase, This is a traditional watercolor and pencil drawing of a young Japanese girl with straight, shoulder-length black hair and bangs. She has large, expressive brown eyes and a gentle smile. She is wearing a white sailor-style school uniform with a dark blue collar and a matching blue ribbon tied in a bow at the front. The background is plain white, highlighting the subject. The texture of the watercolor paper is visible, adding a subtle grainy effect. The artist's signature, \"CHASEREXE,\" is written in red in the bottom left corner. The overall style is soft and delicate, with gentle shading and a focus on simplicity and innocence."
+          - "nanase, This is a watercolor and ink drawing of an anime-style young woman with a short, straight, reddish-pink bob haircut. Her hair has white highlights and she has a slightly melancholic expression with large, soft brown eyes. She is wearing a simple, loose-fitting white tank top that reveals her slender shoulders and upper arms. The background is a plain, off-white color, emphasizing her figure. The artist's signature, written in blue ink, is visible on the left side of the image. The overall style is gentle and delicate, with soft shading and subtle color gradients. The image exudes a quiet, introspective mood."
+        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: "nanase"
+  version: '1.0'