Spaces:

anvilarth
/

Garage

Runtime error

anvilarth commited on Sep 26, 2024

Commit

2287fea

verified ·

1 Parent(s): 955b30c

Update Garage/AugmenterPipeline.py

Files changed (1) hide show

Garage/AugmenterPipeline.py CHANGED Viewed

@@ -63,7 +63,7 @@ class Augmenter:
         self._models["PowerPaint"].to(device)
         self.device = device
     def __call__(self,
                 image: Image.Image,
                 mask: Image.Image,

         self._models["PowerPaint"].to(device)
         self.device = device
+    @spaces.GPU(duration=120)
     def __call__(self,
                 image: Image.Image,
                 mask: Image.Image,