Update handler.py

handler.py

@@ -69,7 +69,7 @@ class EndpointHandler():
         # Define Generator with seed
         self.generator = torch.Generator(device="cpu").manual_seed(3)
         # this command loads the individual model components on GPU on-demand.
-        self.pipe.enable_model_cpu_offload()
+        # self.pipe.enable_model_cpu_offload()
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
         """