Add dependency installation for musubi-tuner in app.py during startup. Implement best-effort upgrade for pip, setuptools, and wheel, and handle optional Torch extras for editable installation. Fallback to plain install if editable fails, ensuring smoother setup process.

app.py

@@ -374,6 +374,41 @@ def _startup_clone_musubi_tuner() -> None:
                 print(f"[QIE] Clone failed in fallback as well: {e2}")
 
 
+def _run_pip(args: List[str], cwd: Optional[str] = None) -> None:
+    cmd = [sys.executable, "-m", "pip"] + args
+    try:
+        print(f"[QIE] pip {' '.join(args)} (cwd={cwd or os.getcwd()})")
+        subprocess.run(cmd, check=True, cwd=cwd)
+    except subprocess.CalledProcessError as e:
+        print(f"[QIE] pip failed: {e}")
+
+
+def _startup_install_musubi_deps() -> None:
+    repo_dir = MUSUBI_TUNER_DIR_RUNTIME
+    if not os.path.isdir(repo_dir):
+        print(f"[QIE] Skip deps: musubi-tuner not found at {repo_dir}")
+        return
+    # Upgrade basic build tooling (best-effort)
+    try:
+        _run_pip(["install", "-U", "pip", "setuptools", "wheel"])
+    except Exception:
+        pass
+
+    # Optional Torch extra via env: MUSUBI_TUNER_TORCH_EXTRA=cu124|cu128
+    extra = os.environ.get("MUSUBI_TUNER_TORCH_EXTRA", "").strip()
+    editable_spec = "." if not extra else f".[{extra}]"
+
+    # Install musubi-tuner in editable mode to expose entrypoints and deps
+    try:
+        _run_pip(["install", "-e", editable_spec], cwd=repo_dir)
+    except Exception:
+        # Fallback: plain install without editable
+        try:
+            _run_pip(["install", editable_spec], cwd=repo_dir)
+        except Exception:
+            print("[QIE] WARN: musubi-tuner installation failed. Continuing.")
+
+
 @spaces.GPU
 def run_training(
     output_name: str,
@@ -694,6 +729,8 @@ def _startup_download_models() -> None:
 if __name__ == "__main__":
     # 1) Ensure musubi-tuner is cloned before anything else
     _startup_clone_musubi_tuner()
+    # 1.1) Install musubi-tuner dependencies (best-effort)
+    _startup_install_musubi_deps()
 
     # 2) Download models at startup (blocking by design)
     _startup_download_models()