xinjie.wang commited on
Commit
4add0d9
·
1 Parent(s): 7c935af
Files changed (1) hide show
  1. common.py +2 -5
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
- print(torch.cuda.memory_allocated() / 1024**2, "MB 已分配")
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]