xinjie.wang commited on
Commit
4091e8b
·
1 Parent(s): adefc17
Files changed (1) hide show
  1. common.py +1 -0
common.py CHANGED
@@ -305,6 +305,7 @@ def get_cached_image(image_path: str) -> Image.Image:
305
  return Image.open(image_path).resize((512, 512))
306
 
307
 
 
308
  def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
309
  return {
310
  "gaussian": {
 
305
  return Image.open(image_path).resize((512, 512))
306
 
307
 
308
+ @spaces.GPU
309
  def pack_state(gs: Gaussian, mesh: MeshExtractResult) -> dict:
310
  return {
311
  "gaussian": {