Spaces:

jingyangcarl
/

matgen

Running

jingyangcarl commited on Jun 6

Commit

75cdf8c

1 Parent(s): e51b78e

debug zerogpu

Files changed (1) hide show

model.py CHANGED Viewed

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