Update SDLens/hooked_sd_pipeline.py

SDLens/hooked_sd_pipeline.py

@@ -31,7 +31,6 @@ class HookedDiffusionAbstractPipeline:
     def from_pretrained(cls, *args, **kwargs):
         return cls(cls.parent_cls.from_pretrained(*args, **kwargs))
 
-    @spaces.GPU
     def run_with_hooks(self, 
         *args,
         position_hook_dict: Dict[str, Union[Callable, List[Callable]]], 
@@ -71,7 +70,6 @@ class HookedDiffusionAbstractPipeline:
         return output
 
     
-    @spaces.GPU
     def run_with_cache(self, 
         *args,
         positions_to_cache: List[str],
@@ -127,7 +125,6 @@ class HookedDiffusionAbstractPipeline:
         return output, cache_dict
 
     
-    @spaces.GPU
     def run_with_hooks_and_cache(self,
         *args,
         position_hook_dict: Dict[str, Union[Callable, List[Callable]]],
@@ -312,6 +309,7 @@ class HookedDiffusionAbstractPipeline:
     def __setattr__(self, name, value):
         return setattr(self.pipe, name, value)
 
+    @spaces.GPU
     def __call__(self, *args, **kwargs):
         return self.pipe(*args, **kwargs)