jiaweir commited on
Commit
c06c03d
β€’
1 Parent(s): eb23d10
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -3
app.py CHANGED
@@ -11,7 +11,7 @@ import spaces
11
 
12
 
13
  subprocess.run(shlex.split("pip install wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl"))
14
- subprocess.run(shlex.split("pip install xformers==0.0.23 --no-deps --index-url https://download.pytorch.org/whl/cu118"))
15
 
16
  from huggingface_hub import hf_hub_download
17
  ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16_fixrot.safetensors")
 
11
 
12
 
13
  subprocess.run(shlex.split("pip install wheels/diff_gaussian_rasterization-0.0.0-cp310-cp310-linux_x86_64.whl"))
14
+ # subprocess.run(shlex.split("pip install xformers==0.0.23 --no-deps --index-url https://download.pytorch.org/whl/cu118"))
15
 
16
  from huggingface_hub import hf_hub_download
17
  ckpt_path = hf_hub_download(repo_id="ashawkey/LGM", filename="model_fp16_fixrot.safetensors")
requirements.txt CHANGED
@@ -11,9 +11,8 @@ imageio
11
  imageio-ffmpeg
12
  omegaconf
13
 
14
- torch==2.1.0 --index-url https://download.pytorch.org/whl/cu118
15
- torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
16
- torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
17
  einops
18
  plyfile
19
  pygltflib
 
11
  imageio-ffmpeg
12
  omegaconf
13
 
14
+ torch==2.0.0
15
+ xformers
 
16
  einops
17
  plyfile
18
  pygltflib