try this

handler.py

@@ -43,7 +43,8 @@ class EndpointHandler():
         logger.info(f"Printing loaded image into library {decoded_image}")
         
         # run inference pipeline
-        upscaled_image = self.pipe(prompt="", image = decoded_image).images[0]
+        with autocast(device.type):
+            upscaled_image = self.pipe(prompt="", image = decoded_image).images[0]
 
         # encode image as base 64
         buffered = BytesIO()