Fix bugs.

webui/runner.py

@@ -62,7 +62,7 @@ class Runner:
             num_inference_steps=num_steps,
             enable_gradient_checkpoint=False,
         )
-        output_image = to_pil_image(result[0])
+        output_image = to_pil_image(result[0].float())
         del result
         torch.cuda.empty_cache()
         return [output_image]
@@ -97,7 +97,7 @@ class Runner:
             num_images_per_prompt=num_images_per_prompt,
             enable_gradient_checkpoint=False
         )
-        output_images = [to_pil_image(image) for image in images]
+        output_images = [to_pil_image(image.float()) for image in images]
 
         del images
         torch.cuda.empty_cache()
@@ -154,7 +154,7 @@ class Runner:
         else:
             raise ValueError
         
-        output_images = [to_pil_image(image) for image in results]
+        output_images = [to_pil_image(image.float()) for image in results]
         del results
         torch.cuda.empty_cache()
         return output_images