xrg
commited on
Commit
·
e47b08e
1
Parent(s):
5b63cd2
fix dependences
Browse files- app.py +1 -0
- requirements.txt +3 -2
app.py
CHANGED
|
@@ -137,6 +137,7 @@ pipe_image_plus = pipe_image_plus.to(device)
|
|
| 137 |
bg_remover = rembg.new_session()
|
| 138 |
|
| 139 |
# process function
|
|
|
|
| 140 |
def process(condition_input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42, mv_moedl_option=None):
|
| 141 |
|
| 142 |
# seed
|
|
|
|
| 137 |
bg_remover = rembg.new_session()
|
| 138 |
|
| 139 |
# process function
|
| 140 |
+
@spaces.GPU
|
| 141 |
def process(condition_input_image, prompt, prompt_neg='', input_elevation=0, input_num_steps=30, input_seed=42, mv_moedl_option=None):
|
| 142 |
|
| 143 |
# seed
|
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
-
torch
|
|
|
|
| 2 |
numpy
|
| 3 |
tyro
|
| 4 |
diffusers
|
|
@@ -34,4 +35,4 @@ opencv-python
|
|
| 34 |
optimum[onnxruntime-gpu]
|
| 35 |
rembg[gpu,cli]
|
| 36 |
httpx[socks]
|
| 37 |
-
|
|
|
|
| 1 |
+
torch==2.0.0
|
| 2 |
+
xformers
|
| 3 |
numpy
|
| 4 |
tyro
|
| 5 |
diffusers
|
|
|
|
| 35 |
optimum[onnxruntime-gpu]
|
| 36 |
rembg[gpu,cli]
|
| 37 |
httpx[socks]
|
| 38 |
+
git+https://github.com/NVlabs/nvdiffrast/
|