Fix No @spaces.GPU function detected during startup when deploy on hugging face

.hf-space

@@ -0,0 +1,12 @@
+title: Hunyuan 3D Generator
+emoji: 🧊
+colorFrom: blue
+colorTo: indigo
+sdk: gradio
+sdk_version: 4.44.1
+app_file: hg_app.py
+pinned: false
+license: apache-2.0
+python_version: 3.10
+hf_oauth: false
+hardware: gpu

hg_app.py

@@ -10,10 +10,24 @@ args = parser.parse_args()
 print(f"Running on {'local' if args.local else 'huggingface'}")
 if not args.local:
     import os
-    import spaces
     import subprocess
     import sys
     import shlex
+    
+    # Try to import spaces, create a fallback if not available
+    try:
+        import spaces
+        print("Successfully imported Hugging Face spaces module")
+    except ImportError:
+        print("WARNING: Could not import Hugging Face spaces module, using fallback implementation")
+        class spaces:
+            class GPU:
+                def __init__(self, duration=60):
+                    self.duration = duration
+                    print(f"Created mock GPU decorator with duration {duration}")
+                def __call__(self, func):
+                    print(f"Applied mock GPU decorator to {func.__name__}")
+                    return func
 
     print("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
     os.system("cd /home/user/app/hy3dgen/texgen/differentiable_renderer/ && bash compile_mesh_painter.sh")
@@ -30,8 +44,10 @@ else:
         class GPU:
             def __init__(self, duration=60):
                 self.duration = duration
+                print(f"Created local mock GPU decorator with duration {duration}")
             def __call__(self, func):
-                return func 
+                print(f"Applied local mock GPU decorator to {func.__name__}")
+                return func
 
 import os
 import shutil
@@ -128,6 +144,7 @@ def _gen_shape(
     check_box_rembg: bool,
     req: gr.Request,
 ):
+    print(f"Starting _gen_shape with GPU acceleration, prompt: '{caption}', seed: {seed}")
     if caption: print('prompt is', caption)
     save_folder = os.path.join(SAVE_DIR, str(req.session_hash)) 
     os.makedirs(save_folder, exist_ok=True)
@@ -192,6 +209,7 @@ def generation_all(
     check_box_rembg: bool,
     req: gr.Request,
 ):
+    print(f"Starting generation_all with GPU acceleration, seed: {seed}")
     mesh, save_folder, image = _gen_shape(
         caption,
         image,
@@ -228,6 +246,7 @@ def shape_generation(
     check_box_rembg: bool,
     req: gr.Request,
 ):
+    print(f"Starting shape_generation with GPU acceleration, seed: {seed}")
     mesh, save_folder, image = _gen_shape(
         caption,
         image,

requirements-hf.txt

@@ -0,0 +1,7 @@
+gradio==4.44.1
+huggingface_hub>=0.16.4
+torch>=2.0.0
+pillow>=9.0.0
+fastapi
+uvicorn
+spaces

run_in_hf.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+echo "Setting up Hunyuan3D-2.0 environment..."
+
+# Install custom dependencies
+cd /home/user/app
+
+# Print diagnostic information
+echo "Python version:"
+python --version
+
+echo "CUDA availability:"
+python -c "import torch; print(f'CUDA available: {torch.cuda.is_available()}')"
+python -c "import torch; print(f'CUDA device count: {torch.cuda.device_count()}')"
+python -c "import torch; print(f'CUDA device name: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else \"N/A\"}')"
+
+# Install the required packages
+pip install -r requirements-hf.txt
+
+# Install the package in development mode
+pip install -e .
+
+echo "Environment setup completed."

setup.py

@@ -28,5 +28,13 @@ setup(
     name="hy3dgen",
     version="2.0.0",
     packages=find_packages(),
-
+    install_requires=[
+        "gradio>=4.0.0",
+        "huggingface_hub>=0.16.4",
+        "torch>=2.0.0",
+        "pillow>=9.0.0",
+        "fastapi",
+        "uvicorn",
+    ],
+    python_requires=">=3.8",
 )