Upload 3 files

app.py

@@ -2,9 +2,6 @@ import gradio as gr
 import numpy as np
 import os
 
-import torch
-print("CUDA_ARCH ==>", torch.cuda.get_device_capability(0))
-
 try:
     from train import *
     print('==> simple-knn & diff-gaussian-rasterization already installed!')
@@ -13,8 +10,6 @@ except:
     # https://github.com/pytorch/extension-cpp/issues/71
     os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0;5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6;8.7;8.9;9.0+PTX"
     print('==> TORCH_CUDA_ARCH_LIST =', os.environ.get('TORCH_CUDA_ARCH_LIST'))
-    os.system("python -m pip install git+https://github.com/openai/point-e.git")
-    print('==> point-e installed!')
     os.system("python -m pip install git+https://github.com/YixunLiang/simple-knn.git")
     print('==> simple-knn installed!')
     os.system("python -m pip install git+https://github.com/YixunLiang/diff-gaussian-rasterization.git")

pre-requirements.txt

@@ -0,0 +1,20 @@
+pip==23.3.1
+torch==2.0.0 --index-url https://download.pytorch.org/whl/cu118
+torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cu118
+torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
+tqdm
+mediapipe
+Pillow
+diffusers==0.18.2
+transformers==4.30.2
+fire==0.5.0
+huggingface_hub==0.16.4
+imageio==2.31.1
+imageio-ffmpeg
+PyYAML
+safetensors
+wandb
+accelerate
+triton
+gradio
+plyfile==0.8.1

requirements.txt

@@ -1,21 +1,4 @@
-pip==24.2
-# torch==2.1.2 --index-url https://download.pytorch.org/whl/cu121
-# torchvision==0.16.2 --index-url https://download.pytorch.org/whl/cu121
-# torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu121
-tqdm==4.66.1
-mediapipe==0.10.8
-Pillow==10.1.0
-diffusers==0.18.2
-transformers==4.30.2
-fire==0.5.0
-huggingface_hub==0.16.4
-imageio==2.31.1
-imageio-ffmpeg==0.4.9
-PyYAML==6.0.1
-safetensors==0.4.1
-wandb==0.16.0
-accelerate==0.24.1
-triton==2.1.0
-gradio==4.7.1
-plyfile==0.8.1
-xformers==0.0.23.post1 --index-url https://download.pytorch.org/whl/cu121
+xformers==0.0.18
+git+https://github.com/openai/point-e.git
+# git+https://github.com/YixunLiang/simple-knn.git
+# git+https://github.com/YixunLiang/diff-gaussian-rasterization.git