Update Garage/models/PowerPaintModel.py
Browse files
Garage/models/PowerPaintModel.py
CHANGED
@@ -114,7 +114,7 @@ class PowerPaintModel:
|
|
114 |
self.pipe.to(device)
|
115 |
self.device = device
|
116 |
|
117 |
-
@spaces.GPU(duration=120)
|
118 |
def __call__(
|
119 |
self,
|
120 |
input_image: Dict[str, Image.Image],
|
|
|
114 |
self.pipe.to(device)
|
115 |
self.device = device
|
116 |
|
117 |
+
# @spaces.GPU(duration=120)
|
118 |
def __call__(
|
119 |
self,
|
120 |
input_image: Dict[str, Image.Image],
|