update

app.py

@@ -66,29 +66,29 @@ def images_to_video(images, output_path, fps=30):
 
 import shutil
 
-def find_cuda():
-    # Check if CUDA_HOME or CUDA_PATH environment variables are set
-    cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
-
-    if cuda_home and os.path.exists(cuda_home):
-        return cuda_home
-
-    # Search for the nvcc executable in the system's PATH
-    nvcc_path = shutil.which('nvcc')
-
-    if nvcc_path:
-        # Remove the 'bin/nvcc' part to get the CUDA installation path
-        cuda_path = os.path.dirname(os.path.dirname(nvcc_path))
-        return cuda_path
-
-    return None
-
-cuda_path = find_cuda()
-
-if cuda_path:
-    print(f"CUDA installation found at: {cuda_path}")
-else:
-    print("CUDA installation not found")
+#def find_cuda():
+#    # Check if CUDA_HOME or CUDA_PATH environment variables are set
+#    cuda_home = os.environ.get('CUDA_HOME') or os.environ.get('CUDA_PATH')
+#
+#    if cuda_home and os.path.exists(cuda_home):
+#        return cuda_home
+#
+#    # Search for the nvcc executable in the system's PATH
+#    nvcc_path = shutil.which('nvcc')
+#
+#    if nvcc_path:
+#        # Remove the 'bin/nvcc' part to get the CUDA installation path
+#        cuda_path = os.path.dirname(os.path.dirname(nvcc_path))
+#        return cuda_path
+#
+#    return None
+#
+#cuda_path = find_cuda()
+#
+#if cuda_path:
+#    print(f"CUDA installation found at: {cuda_path}")
+#else:
+#    print("CUDA installation not found")
 
 config_path = 'configs/instant-mesh-large.yaml'
 config = OmegaConf.load(config_path)
@@ -98,7 +98,13 @@ infer_config = config.infer_config
 
 IS_FLEXICUBES = True if config_name.startswith('instant-mesh') else False
 
-device = torch.device('cuda')
+if torch.backends.mps.is_available():
+    _device = "mps"
+elif torch.cuda.is_available():
+    _device = "cuda"
+else:
+    _device = "cpu"
+device = torch.device(_device)
 
 # load diffusion model
 print('Loading diffusion model ...')
@@ -358,4 +364,4 @@ with gr.Blocks() as demo:
         outputs=[output_model_obj]
     )
 
-demo.launch()
+demo.launch()

requirements.txt

@@ -1,6 +1,6 @@
-torch==2.1.0
-torchvision==0.16.0
-torchaudio==2.1.0
+#torch==2.1.0
+#torchvision==0.16.0
+#torchaudio==2.1.0
 pytorch-lightning==2.1.2
 einops
 omegaconf
@@ -18,6 +18,6 @@ bitsandbytes
 imageio[ffmpeg]
 xatlas
 plyfile
-xformers==0.0.22.post7
+#xformers==0.0.22.post7
 git+https://github.com/NVlabs/nvdiffrast/
-huggingface-hub
+huggingface-hub