Spaces:
Running
on
Zero
Running
on
Zero
xinjie.wang
commited on
Commit
·
4add0d9
1
Parent(s):
7c935af
update
Browse files
common.py
CHANGED
@@ -160,7 +160,6 @@ def render_mesh(sample, extrinsics, intrinsics, options={}, **kwargs):
|
|
160 |
return rets
|
161 |
|
162 |
|
163 |
-
@spaces.GPU
|
164 |
def render_video(
|
165 |
sample,
|
166 |
resolution=512,
|
@@ -188,9 +187,7 @@ def render_video(
|
|
188 |
{"resolution": resolution, "bg_color": bg_color},
|
189 |
**kwargs,
|
190 |
)
|
191 |
-
|
192 |
-
print(torch.cuda.memory_reserved() / 1024**2, "MB 已预留")
|
193 |
-
|
194 |
return result
|
195 |
|
196 |
|
@@ -373,7 +370,7 @@ def image_to_3d(
|
|
373 |
},
|
374 |
)
|
375 |
# Set to cpu for memory saving.
|
376 |
-
PIPELINE.cpu()
|
377 |
|
378 |
gs_model = outputs["gaussian"][0]
|
379 |
mesh_model = outputs["mesh"][0]
|
|
|
160 |
return rets
|
161 |
|
162 |
|
|
|
163 |
def render_video(
|
164 |
sample,
|
165 |
resolution=512,
|
|
|
187 |
{"resolution": resolution, "bg_color": bg_color},
|
188 |
**kwargs,
|
189 |
)
|
190 |
+
|
|
|
|
|
191 |
return result
|
192 |
|
193 |
|
|
|
370 |
},
|
371 |
)
|
372 |
# Set to cpu for memory saving.
|
373 |
+
# PIPELINE.cpu()
|
374 |
|
375 |
gs_model = outputs["gaussian"][0]
|
376 |
mesh_model = outputs["mesh"][0]
|