Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -67,6 +67,7 @@ def unpack_state(state: dict) -> Tuple[Gaussian, edict, str]:
|
|
67 |
def get_seed(randomize_seed: bool, seed: int) -> int:
|
68 |
return np.random.randint(0, MAX_SEED) if randomize_seed else seed
|
69 |
|
|
|
70 |
def text_to_3d(
|
71 |
prompt: str,
|
72 |
seed: int,
|
@@ -99,6 +100,7 @@ def text_to_3d(
|
|
99 |
torch.cuda.empty_cache()
|
100 |
return state, video_path
|
101 |
|
|
|
102 |
def extract_glb(
|
103 |
state: dict,
|
104 |
mesh_simplify: float,
|
|
|
67 |
def get_seed(randomize_seed: bool, seed: int) -> int:
|
68 |
return np.random.randint(0, MAX_SEED) if randomize_seed else seed
|
69 |
|
70 |
+
@spaces.GPU
|
71 |
def text_to_3d(
|
72 |
prompt: str,
|
73 |
seed: int,
|
|
|
100 |
torch.cuda.empty_cache()
|
101 |
return state, video_path
|
102 |
|
103 |
+
@spaces.GPU
|
104 |
def extract_glb(
|
105 |
state: dict,
|
106 |
mesh_simplify: float,
|