Deploy BASILISC worker: app.py

app.py

@@ -21,6 +21,36 @@ from PIL import Image
 from torchvision import transforms
 from transformers import AutoModelForImageSegmentation
 
+# Keep the Space image build simple: fetch a pinned MV-Adapter source snapshot
+# at runtime instead of asking pip to resolve the entire upstream package and
+# its dependency tree during the Hugging Face image build.
+MV_ADAPTER_COMMIT = "4277e0018232bac82bb2c103caf0893cedb711be"
+MV_ADAPTER_ROOT = Path("/tmp/basilisc-mv-adapter")
+
+def _ensure_mvadapter_source():
+    pkg = MV_ADAPTER_ROOT / "mvadapter"
+    if not pkg.is_dir():
+        import urllib.request
+        import zipfile
+        archive = Path("/tmp/basilisc-mv-adapter.zip")
+        url = f"https://codeload.github.com/huanngzh/MV-Adapter/zip/{MV_ADAPTER_COMMIT}"
+        print(f"Fetching pinned MV-Adapter source: {MV_ADAPTER_COMMIT}", flush=True)
+        urllib.request.urlretrieve(url, archive)
+        extract_root = Path("/tmp/basilisc-mv-adapter-extract")
+        shutil.rmtree(extract_root, ignore_errors=True)
+        extract_root.mkdir(parents=True, exist_ok=True)
+        with zipfile.ZipFile(archive, "r") as zf:
+            zf.extractall(extract_root)
+        children = [x for x in extract_root.iterdir() if x.is_dir()]
+        if len(children) != 1 or not (children[0] / "mvadapter").is_dir():
+            raise RuntimeError("Downloaded MV-Adapter archive had an unexpected layout")
+        shutil.rmtree(MV_ADAPTER_ROOT, ignore_errors=True)
+        shutil.move(str(children[0]), str(MV_ADAPTER_ROOT))
+    if str(MV_ADAPTER_ROOT) not in sys.path:
+        sys.path.insert(0, str(MV_ADAPTER_ROOT))
+
+_ensure_mvadapter_source()
+
 from mvadapter.models.attention_processor import DecoupledMVRowColSelfAttnProcessor2_0
 from mvadapter.pipelines.pipeline_mvadapter_i2mv_sdxl import MVAdapterI2MVSDXLPipeline
 from mvadapter.schedulers.scheduling_shift_snr import ShiftSNRScheduler