Pre-download Gemma at startup; bump guidance to 5.0 for prompt adherence

app.py

@@ -91,13 +91,21 @@ def setup():
     for repo, fname, dest in SULPHUR_ASSETS + LTX_ASSETS:
         _download(repo, fname, dest)
 
-    te_dir = CKPTS_DIR / "umt5-xxl"
-    if not te_dir.exists():
-        from huggingface_hub import snapshot_download
-        print("[download] umt5-xxl...")
-        snapshot_download(repo_id="DeepBeepMeep/Wan2.1",
-                          allow_patterns=["umt5-xxl/*"],
-                          local_dir=str(CKPTS_DIR), token=_HF_TOKEN)
+    # Gemma text encoder — must stay in its subfolder (Wan2GP looks there by name)
+    _gemma_folder = "gemma-3-12b-it-qat-q4_0-unquantized"
+    _gemma_file = f"{_gemma_folder}_quanto_bf16_int8.safetensors"
+    gemma_dest = CKPTS_DIR / _gemma_folder / _gemma_file
+    if not gemma_dest.exists():
+        from huggingface_hub import hf_hub_download
+        print("[download] Gemma text encoder...")
+        hf_hub_download(
+            repo_id="DeepBeepMeep/LTX-2",
+            filename=f"{_gemma_folder}/{_gemma_file}",
+            local_dir=str(CKPTS_DIR),
+            token=_HF_TOKEN,
+        )
+    else:
+        print("[download] cached: Gemma text encoder")
 
     FINETUNES_DIR.mkdir(parents=True, exist_ok=True)
     (FINETUNES_DIR / "sulphur_2_base.json").write_text(json.dumps(SULPHUR_FINETUNE, indent=2))
@@ -178,7 +186,7 @@ with gr.Blocks(title="Sulphur — Image to Video") as demo:
             with gr.Accordion("Advanced", open=False):
                 resolution_dd = gr.Dropdown(RESOLUTIONS, value="832x480", label="Resolution")
                 steps_sl      = gr.Slider(1, 50, value=8,   step=1,   label="Steps")
-                guidance_sl   = gr.Slider(1.0, 10.0, value=3.5, step=0.5, label="Guidance Scale")
+                guidance_sl   = gr.Slider(1.0, 10.0, value=5.0, step=0.5, label="Guidance Scale")
                 frames_sl     = gr.Slider(17, 257, value=81, step=8,  label="Frames")
                 seed_num      = gr.Number(value=-1, label="Seed (-1 = random)", precision=0)
             run_btn = gr.Button("Generate", variant="primary")

generate.py

@@ -17,7 +17,7 @@ MODEL_SHORTHANDS = {
 DEFAULTS = {
     "sulphur_2_base": {
         "num_inference_steps": 8,
-        "guidance_scale":      3.5,
+        "guidance_scale":      5.0,
         "resolution":          "832x480",
         "video_length":        81,
     },