Spaces:
Runtime error
Runtime error
Commit
·
c0aff90
1
Parent(s):
5d94b0f
update space
Browse files
app.py
CHANGED
|
@@ -178,7 +178,7 @@ def main():
|
|
| 178 |
print(state)
|
| 179 |
return out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4
|
| 180 |
|
| 181 |
-
@spaces.GPU()
|
| 182 |
@torch.no_grad()
|
| 183 |
def generate_3d_with_shap_e(prompt, rand_seed, state):
|
| 184 |
print("Check if I can use partial")
|
|
@@ -234,7 +234,7 @@ def main():
|
|
| 234 |
del latents
|
| 235 |
return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
|
| 236 |
|
| 237 |
-
@spaces.GPU()
|
| 238 |
@torch.no_grad()
|
| 239 |
def _3d_editing(instruction, rand_seed, state):
|
| 240 |
set_seed(rand_seed)
|
|
|
|
| 178 |
print(state)
|
| 179 |
return out_gen_1, out_gen_2, out_gen_3, out_gen_4, edited_1, edited_2, edited_3, edited_4
|
| 180 |
|
| 181 |
+
@spaces.GPU(duration=180)
|
| 182 |
@torch.no_grad()
|
| 183 |
def generate_3d_with_shap_e(prompt, rand_seed, state):
|
| 184 |
print("Check if I can use partial")
|
|
|
|
| 234 |
del latents
|
| 235 |
return mesh_path[0], mesh_path[1], mesh_path[2], mesh_path[3], state
|
| 236 |
|
| 237 |
+
@spaces.GPU(duration=180)
|
| 238 |
@torch.no_grad()
|
| 239 |
def _3d_editing(instruction, rand_seed, state):
|
| 240 |
set_seed(rand_seed)
|