--- job: extension config: # this name will be the folder and filename name name: "0__11Xx" process: - type: 'sd_trainer' # root folder to save training sessions/samples/weights training_folder: "/opt/stationthis/jobs/training-1774557949786/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: "0__11Xx" 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-1774557949786/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 - "0__11Xx, This is a digital artwork featuring a stylized, surreal composition. At the center, a person with short, light brown hair, wearing a white jacket and black pants, stands on a grassy hill. They face away from the viewer, gazing at a vivid sunset over a coastal cityscape. The sky is a gradient of pink, orange, and purple hues, with scattered clouds and flying birds. A large, circular frame encloses the sunset, contrasting with the deep blue background. Below, palm trees and city lights are visible, adding depth. The artwork uses bold colors and a mix of realistic and abstract styles, creating a dreamlike atmosphere." - "0__11Xx, This digital artwork depicts a person with shoulder-length, light brown hair sitting cross-legged on the floor. The figure is viewed from behind and wears a white, long-sleeved shirt with black straps and black, thigh-high socks. The background is a muted beige with abstract, white rectangular shapes and a dark plant silhouette to the left. The person's silhouette is partially overlaid with a textured, grungy effect, adding a layer of depth. The artist's signature, \"@TITOMORU,\" is visible in the bottom right corner. The overall style is minimalist with a blend of abstract and grunge elements, emphasizing contrast and texture." - "0__11Xx, This is a digital illustration featuring a mysterious, ethereal scene. A small, silhouette-like figure with short, purple hair stands at the center, wearing a flowing white coat and white pants. The figure faces away, gazing into a swirling, vortex-like void surrounded by a galaxy of vibrant, multicolored stars and nebulas. The dark background contrasts sharply with the bright, cosmic swirl, creating a sense of depth and otherworldliness. The texture of the swirl is rich and painterly, with swirling, brushstroke-like patterns. The overall style is surreal and dreamlike, evoking a sense of wonder and mystery. The image is highly detailed and evocative." - "0__11Xx, This is a digital anime-style drawing featuring a solitary female figure standing with her back to the viewer. She has long, dark green hair that partially obscures her face and wears a black jacket with a white collar, a short white skirt, and black thigh-high stockings. Her silhouette is partially pixelated, especially around the upper back and shoulders. The background depicts a vibrant blue sky with fluffy white clouds, a vast ocean, and a distant cityscape with pixelated buildings at the bottom. The overall color palette is dominated by blue and white, with a stark contrast between the dark figure and the bright background." 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: "0__11Xx" version: '1.0'