Spaces:
Runtime error
Runtime error
pengHTYX
commited on
Commit
•
db2690c
1
Parent(s):
b20a911
'test'
Browse files- app.py +1 -1
- 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 |
-
'
|
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
|
|