pengHTYX commited on
Commit
db2690c
1 Parent(s): b20a911
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +5 -2
app.py CHANGED
@@ -157,7 +157,7 @@ def load_era3d_pipeline(cfg):
157
  # Load scheduler, tokenizer and models.
158
 
159
  pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
160
- '../MacLab-Era3D-512-6view',
161
  torch_dtype=weight_dtype
162
  )
163
 
 
157
  # Load scheduler, tokenizer and models.
158
 
159
  pipeline = StableUnCLIPImg2ImgPipeline.from_pretrained(
160
+ 'pengHTYX/MacLab-Era3D-512-6view',
161
  torch_dtype=weight_dtype
162
  )
163
 
requirements.txt CHANGED
@@ -1,6 +1,10 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  diffusers[torch]==0.26.0
3
  transformers==4.37.2
 
 
 
 
4
  bitsandbytes==0.35.4
5
  decord==0.6.0
6
  click==8.1.7
@@ -28,5 +32,4 @@ rembg
28
  segment_anything
29
  gradio==3.50.2
30
  moviepy
31
- kornia
32
- huggingface_hub
 
1
  --extra-index-url https://download.pytorch.org/whl/cu118
2
  diffusers[torch]==0.26.0
3
  transformers==4.37.2
4
+ torch==2.1.2
5
+ torchvision==0.16.2
6
+ torchaudio==2.1.2
7
+ xformers==0.0.25
8
  bitsandbytes==0.35.4
9
  decord==0.6.0
10
  click==8.1.7
 
32
  segment_anything
33
  gradio==3.50.2
34
  moviepy
35
+ kornia