fix: device mixing error when --gpu-only, --fp8... is used

2023-12-19 16:55:30 +09:00
parent 548b2a3542
commit f3c44a568f
2 changed files with 4 additions and 1 deletions
@@ -2,7 +2,7 @@ import configparser
 import os


-version = "V4.48.2"
+version = "V4.48.3"

 dependency_version = 19

@@ -294,6 +294,9 @@ def enhance_detail(image, model, clip, vae, guide_size, guide_size_for_bbox, max
    # downscale
    refined_image = tensor_resize(refined_image, w, h)

+    # prevent mixing of device
+    refined_image = refined_image.cpu()
+
    # don't convert to latent - latent break image
    # preserving pil is much better
    return refined_image, cnet_pil