Spaces:

LTT
/

Kiss3DGen

Running on Zero

JiantaoLin commited on Dec 21, 2024

Commit

697dd0c

1 Parent(s): a45ca60

new

Files changed (1) hide show

pipeline/kiss3d_wrapper.py CHANGED Viewed

@@ -222,8 +222,9 @@ class kiss3d_wrapper(object):
     def context(self):
         if self.config['use_zero_gpu']:
-            import spaces
-            return spaces.GPU()
         else:
             return torch.no_grad()

     def context(self):
         if self.config['use_zero_gpu']:
+            # import spaces
+            # return spaces.GPU()
+            pass
         else:
             return torch.no_grad()