Alias quantized Gemma weights for LTX loader

app.py

@@ -165,6 +165,19 @@ def _download(repo_id, filename, local_dir):
     return path
 
 
+def _ensure_gemma_model_alias(gemma_weight):
+    gemma_weight = Path(gemma_weight)
+    alias = gemma_weight.with_name("model.safetensors")
+    if alias.exists():
+        return alias
+    try:
+        alias.symlink_to(gemma_weight.name)
+    except OSError:
+        os.link(gemma_weight, alias)
+    print(f"[download] ready: {alias}")
+    return alias
+
+
 def download_assets():
     checkpoint = _download("SulphurAI/Sulphur-2-base", "sulphur_distil_bf16.safetensors", ASSETS_DIR)
     sulphur_lora = _download(
@@ -197,8 +210,13 @@ def download_assets():
         f"{gemma_folder}/tokenizer.model",
         f"{gemma_folder}/tokenizer_config.json",
     ]
+    gemma_weight = None
     for filename in gemma_files:
-        _download("DeepBeepMeep/LTX-2", filename, ASSETS_DIR)
+        path = _download("DeepBeepMeep/LTX-2", filename, ASSETS_DIR)
+        if filename.endswith(".safetensors"):
+            gemma_weight = path
+
+    _ensure_gemma_model_alias(gemma_weight)
 
     return {
         "checkpoint": checkpoint,