OnnxStableDiffusionUpscalePipeline, enable hires for onnx.

2026-09-03 11:30:46 +02:00 · 2024-02-02 18:22:22 +09:00
parent df7564a01f
commit 1eaa40678a
12 changed files with 1425 additions and 1103 deletions
@@ -411,6 +411,8 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
            else:
                shared.log.debug(f'Generated: frames={len(output.frames[0])}')
            output.images = output.frames[0]
+        if isinstance(output.images, np.ndarray):
+            output.images = torch.from_numpy(output.images)
    except AssertionError as e:
        shared.log.info(e)
    except ValueError as e: