Update app.py

app.py

@@ -274,51 +274,45 @@ def update_pricing(steps, oauth_token: Union[gr.OAuthToken, None]):
     else:
         return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
 
-config_yaml = {
-    "device": "cuda:0",
-    "network": {
-        "type": "lora",
-        "linear": 16,
-        "linear_alpha": 16
-    },
-    "save": {
-        "dtype": "float16",
-        "save_every": 10000,
-        "max_step_saves_to_keep": 4,
-        "push_to_hub": True,
-        "hf_private": True
-    },
-    "train": {
-        "batch_size": 1,
-        "gradient_accumulation_steps": 1,
-        "train_unet": True,
-        "train_text_encoder": False,
-        "gradient_checkpointing": True,
-        "noise_scheduler": "flowmatch",
-        "optimizer": "adamw8bit",
-        "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": 1000,
-        "width": 1024,
-        "height": 1024,
-        "neg": "",
-        "seed": 42,
-        "walk_seed": True,
-        "guidance_scale": 3.5,
-        "sample_steps": 28
-    }
-}
+config_yaml = '''
+device: cuda:0
+model:
+  is_flux: true
+  name_or_path: black-forest-labs/FLUX.1-dev
+  quantize: true
+network:
+  linear: 16 #it will overcome the 'rank' parameter
+  linear_alpha: 16 #you can have an alpha different than the ranking if you'd like
+  type: lora
+sample:
+  guidance_scale: 3.5
+  height: 1024
+  neg: '' #doesn't work for FLUX
+  sample_every: 1000
+  sample_steps: 28
+  sampler: flowmatch
+  seed: 42
+  walk_seed: true
+  width: 1024
+save:
+  dtype: float16
+  hf_private: true
+  max_step_saves_to_keep: 4
+  push_to_hub: true
+  save_every: 10000
+train:
+  batch_size: 1
+  dtype: bf16
+  ema_config:
+    ema_decay: 0.99
+    use_ema: true
+  gradient_accumulation_steps: 1
+  gradient_checkpointing: true
+  noise_scheduler: flowmatch 
+  optimizer: adamw8bit #options: prodigy, dadaptation, adamw, adamw8bit, lion, lion8bit
+  train_text_encoder: false #probably doesn't work for flux
+  train_unet: true
+'''
 
 theme = gr.themes.Monochrome(
     text_size=gr.themes.Size(lg="18px", md="15px", sm="13px", xl="22px", xs="12px", xxl="24px", xxs="9px"),