Update handler.py

handler.py

@@ -16,8 +16,8 @@ class EndpointHandler():
     def __init__(self, path=""):
         self.pipeline = DiffusionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages", torch_dtype=dtype).to(device)
         
-        # this command loads the individual model components on GPU on-demand.
-        self.pipeline.enable_model_cpu_offload()
+        # # this command loads the individual model components on GPU on-demand.
+        # self.pipeline.enable_model_cpu_offload()
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         image = data.pop("image", None)