Spaces:

jingyangcarl
/

matgen

Sleeping

jingyangcarl commited on Jun 6

Commit

52e266e

1 Parent(s): ee9e7ab

debug zerogpu

Files changed (1) hide show

model.py CHANGED Viewed

@@ -101,7 +101,7 @@ class Model:
     def get_prompt(self, prompt: str, additional_prompt: str) -> str:
         return additional_prompt if not prompt else f"{prompt}, {additional_prompt}"
-    @spaces.GPU #[uncomment to use ZeroGPU]
     @torch.autocast("cuda")
     def run_pipe(
         self,

     def get_prompt(self, prompt: str, additional_prompt: str) -> str:
         return additional_prompt if not prompt else f"{prompt}, {additional_prompt}"
+    # @spaces.GPU #[uncomment to use ZeroGPU]
     @torch.autocast("cuda")
     def run_pipe(
         self,