asoderznik
/

sdx4-upscaler

StableDiffusionUpscalePipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

asoderznik commited on Aug 29, 2023

Commit

af3699e

•

1 Parent(s): 87d9a6a

try this

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -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]
         # encode image as base 64
         buffered = BytesIO()

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