Spaces:

jingyangcarl
/

matgen

Running

jingyangcarl commited on Jun 6

Commit

8918c2e

1 Parent(s): fd665d1

debug zerogpu

Files changed (1) hide show

model.py CHANGED Viewed

@@ -131,6 +131,7 @@ class Model:
             image=control_image,
         ).images
     @torch.inference_mode()
     def process_canny(
         self,

             image=control_image,
         ).images
+    @spaces.GPU #[uncomment to use ZeroGPU]
     @torch.inference_mode()
     def process_canny(
         self,