Upload train_lora_flux_24gb.yaml with huggingface_hub

train_lora_flux_24gb.yaml

@@ -21,7 +21,7 @@ config:
         linear_alpha: 16
       save:
         dtype: float16 # precision to save
-        save_every: 250 # save every this many steps
+        save_every: 500 # save every this many steps
         max_step_saves_to_keep: 4 # 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         
@@ -42,14 +42,14 @@ config:
           resolution: [512, 768, 1024] # flux enjoys multiple resolutions
       train:
         batch_size: 1
-        steps: 2000 # total number of steps to train 500 - 4000 is a good range
+        steps: 3000 # 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
+        lr: 0.0001
         #        skip_first_sample: true
         # uncomment to completely disable sampling
         #        disable_sampling: true
@@ -66,11 +66,11 @@ config:
         # huggingface model name or path
         name_or_path: "black-forest-labs/FLUX.1-dev"
         is_flux: true
-        quantize: true # run 8bit mixed precision
+        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
+        sample_every: 500 # sample every this many steps
         width: 1024
         height: 1024
         prompts:
@@ -92,6 +92,7 @@ config:
         walk_seed: true
         guidance_scale: 4
         sample_steps: 20
+      trigger_word: p3r5on
 # you can add any additional meta info here. [name] is replaced with config name at top
 meta:
   name: "[name]"