asoderznik
/

sdx4-upscaler

StableDiffusionUpscalePipeline

stable-diffusion

Model card Files Files and versions

asoderznik commited on Aug 30, 2023

Commit

ff2ef5c

·

1 Parent(s): 732c9a5

Update handler.py

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -46,8 +46,8 @@ class EndpointHandler():
         image = Image.open(image_bytes)
         prompt = inputs['prompt']
         logger.info('image')
-        with autocast(device.type):
-            upscaled_image = self.pipe(prompt, image).images[0]
         buffered = BytesIO()
         upscaled_image.save(buffered, format="JPEG")

         image = Image.open(image_bytes)
         prompt = inputs['prompt']
         logger.info('image')
+        upscaled_image = self.pipe(prompt, image).images[0]
         buffered = BytesIO()
         upscaled_image.save(buffered, format="JPEG")