Spaces:
Running
on
Zero
Running
on
Zero
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
|
|
|