Deploy Zoo3D with Zero GPU support

check_zerogpu_ready.sh

@@ -0,0 +1,115 @@
+#!/bin/bash
+
+echo "🔍 Zero GPU Deployment Readiness Check"
+echo "======================================"
+
+# Color codes
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+# Check if torch is in requirements.txt
+echo -n "Checking requirements.txt for torch... "
+if grep -q "^torch==" requirements.txt 2>/dev/null; then
+    echo -e "${RED}❌ FAIL${NC}"
+    echo "  ERROR: torch found in requirements.txt - MUST be removed!"
+    exit 1
+else
+    echo -e "${GREEN}✅ PASS${NC}"
+fi
+
+# Check if torchvision is in requirements.txt
+echo -n "Checking requirements.txt for torchvision... "
+if grep -q "^torchvision==" requirements.txt 2>/dev/null; then
+    echo -e "${RED}❌ FAIL${NC}"
+    echo "  ERROR: torchvision found in requirements.txt - MUST be removed!"
+    exit 1
+else
+    echo -e "${GREEN}✅ PASS${NC}"
+fi
+
+# Check if README_HF.md exists
+echo -n "Checking for README_HF.md... "
+if [ -f "README_HF.md" ]; then
+    echo -e "${GREEN}✅ EXISTS${NC}"
+
+    # Check for python_version in README
+    echo -n "Checking python_version in README_HF.md... "
+    if grep -q "python_version:" README_HF.md; then
+        echo -e "${GREEN}✅ FOUND${NC}"
+    else
+        echo -e "${YELLOW}⚠️  MISSING${NC}"
+        echo "  WARNING: python_version not specified in README_HF.md"
+    fi
+else
+    echo -e "${RED}❌ MISSING${NC}"
+    echo "  ERROR: README_HF.md not found!"
+    exit 1
+fi
+
+# Check if packages.txt exists
+echo -n "Checking for packages.txt... "
+if [ -f "packages.txt" ]; then
+    echo -e "${GREEN}✅ EXISTS${NC}"
+else
+    echo -e "${YELLOW}⚠️  MISSING${NC}"
+    echo "  WARNING: packages.txt not found (may be needed for system dependencies)"
+fi
+
+# Check if app.py exists
+echo -n "Checking for app.py... "
+if [ -f "app.py" ]; then
+    echo -e "${GREEN}✅ EXISTS${NC}"
+else
+    echo -e "${RED}❌ MISSING${NC}"
+    echo "  ERROR: app.py not found!"
+    exit 1
+fi
+
+# Check if mvp.py exists
+echo -n "Checking for mvp.py... "
+if [ -f "mvp.py" ]; then
+    echo -e "${GREEN}✅ EXISTS${NC}"
+
+    # Check for spaces.GPU decorators
+    echo -n "Checking for @spaces.GPU decorators... "
+    if grep -q "@spaces.GPU" mvp.py; then
+        count=$(grep -c "@spaces.GPU" mvp.py)
+        echo -e "${GREEN}✅ FOUND ($count decorators)${NC}"
+    else
+        echo -e "${YELLOW}⚠️  NOT FOUND${NC}"
+        echo "  WARNING: No @spaces.GPU decorators found"
+    fi
+else
+    echo -e "${RED}❌ MISSING${NC}"
+    echo "  ERROR: mvp.py not found!"
+    exit 1
+fi
+
+# Check gradio version
+echo -n "Checking Gradio version... "
+if grep -q "gradio==" requirements.txt; then
+    version=$(grep "gradio==" requirements.txt | cut -d'=' -f3)
+    major=$(echo $version | cut -d'.' -f1)
+    if [ "$major" -ge "4" ]; then
+        echo -e "${GREEN}✅ $version (>=4.0)${NC}"
+    else
+        echo -e "${RED}❌ $version (<4.0)${NC}"
+        echo "  ERROR: Gradio version must be 4.0 or higher"
+    fi
+else
+    echo -e "${YELLOW}⚠️  NOT SPECIFIED${NC}"
+fi
+
+echo ""
+echo "======================================"
+echo -e "${GREEN}✅ All critical checks passed!${NC}"
+echo ""
+echo "Next steps:"
+echo "1. Copy files to your HF Space repository"
+echo "2. Use README_HF.md as your README.md"
+echo "3. Push to Hugging Face"
+echo "4. Enable Zero GPU in Space settings"
+echo ""
+echo "For detailed instructions, see ZERO_GPU_FINAL_INSTRUCTIONS.md"

mvp.py

@@ -6,8 +6,28 @@
 
 import os
 import cv2
-import torch
 import numpy as np
+
+# Import torch - will be provided by Zero GPU environment
+try:
+    import torch
+    TORCH_AVAILABLE = True
+except ImportError:
+    print("⚠️ PyTorch not found - will be installed by Zero GPU environment")
+    TORCH_AVAILABLE = False
+    # For local testing without torch
+    class DummyTorch:
+        cuda = type('cuda', (), {
+            'is_available': lambda: False,
+            'empty_cache': lambda: None,
+            'get_device_capability': lambda: (0, 0),
+            'get_device_name': lambda: 'No GPU'
+        })()
+        def no_grad(self): return self
+        def __enter__(self): return self
+        def __exit__(self, *args): pass
+    if not TORCH_AVAILABLE:
+        torch = DummyTorch()
 import gradio as gr
 import sys
 import shutil
@@ -86,14 +106,29 @@ cropformer_runner.make_cropformer_dir(MK_PATH)
 WORK_DIR = os.environ.get("ZOO3D_WORKDIR", os.path.join(tempfile.gettempdir(), "zoo3d"))
 os.makedirs(WORK_DIR, exist_ok=True)
 from visual_util import predictions_to_glb
-from vggt.models.vggt import VGGT
-from vggt.utils.load_fn import load_and_preprocess_images
-from vggt.utils.pose_enc import pose_encoding_to_extri_intri
-from vggt.utils.geometry import unproject_depth_map_to_point_map
 
-device = "cuda" if torch.cuda.is_available() else 'cpu'
-
-print(f"Using device: {device}")
+# Import VGGT modules only if torch is available
+if TORCH_AVAILABLE:
+    try:
+        from vggt.models.vggt import VGGT
+        from vggt.utils.load_fn import load_and_preprocess_images
+        from vggt.utils.pose_enc import pose_encoding_to_extri_intri
+        from vggt.utils.geometry import unproject_depth_map_to_point_map
+        VGGT_AVAILABLE = True
+    except ImportError as e:
+        print(f"⚠️ VGGT modules not available: {e}")
+        VGGT_AVAILABLE = False
+else:
+    VGGT_AVAILABLE = False
+    print("⚠️ Skipping VGGT imports (PyTorch not available)")
+
+# Set device based on torch availability
+if TORCH_AVAILABLE:
+    device = "cuda" if torch.cuda.is_available() else 'cpu'
+else:
+    device = "cpu"
+
+print(f"Using device: {device} (PyTorch available: {TORCH_AVAILABLE})")
 
 # CPU debug / compatibility knobs:
 # - On CPU, VGGT-1B inference is usually impractical. For debugging, we fall back to a lightweight
@@ -123,6 +158,10 @@ class ModelManager:
     def get_vggt_model(self):
         """Load VGGT model only when needed"""
         if self._vggt_model is None:
+            if not TORCH_AVAILABLE or not VGGT_AVAILABLE:
+                print("⚠️ VGGT model cannot be loaded (PyTorch/VGGT not available)")
+                return None
+
             if not torch.cuda.is_available():
                 if not ZOO3D_ALLOW_CPU:
                     raise RuntimeError("CUDA недоступна. Для этого Space нужен GPU (CUDA).")
@@ -149,6 +188,10 @@ class ModelManager:
     def get_metric3d_model(self):
         """Load Metric3D model only when needed"""
         if self._metric3d_model is None:
+            if not TORCH_AVAILABLE:
+                print("⚠️ Metric3D model cannot be loaded (PyTorch not available)")
+                return None
+
             if not torch.cuda.is_available():
                 return None
 
@@ -187,7 +230,7 @@ class ModelManager:
 
     def clear_cache(self):
         """Clear GPU cache after processing"""
-        if torch.cuda.is_available():
+        if TORCH_AVAILABLE and torch.cuda.is_available():
             torch.cuda.empty_cache()
         gc.collect()
 

requirements.txt

@@ -1,6 +1,5 @@
-# PyTorch - using version compatible with Zero GPU
-torch==2.2.0
-torchvision==0.17.0
+# PyTorch and torchvision are automatically installed by Zero GPU
+# DO NOT specify torch/torchvision here - it will cause conflicts!
 numpy==1.26.1
 Pillow
 huggingface_hub