From 3978346ea5d9edcaadb6c748c2ecfadd4d850e0a Mon Sep 17 00:00:00 2001 From: lvmin Date: Fri, 21 Apr 2023 00:27:30 -0700 Subject: [PATCH] try fix strange apple problem --- scripts/controlnet.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/scripts/controlnet.py b/scripts/controlnet.py index 7a9d7a1..5467ffa 100644 --- a/scripts/controlnet.py +++ b/scripts/controlnet.py @@ -653,10 +653,17 @@ class Script(scripts.Script): detected_map = HWC3(detected_map) def get_pytorch_control(x): - y = x.copy() + # A very safe method to make sure that Apple/Mac works + y = x + + # below is very boring but do not change these. If you change these Apple or Mac may fail. + y = y.copy() + y = y.ascontiguousarray() + y = y.copy() y = torch.from_numpy(y) y = y.float() / 255.0 y = rearrange(y, 'h w c -> c h w') + y = y.clone() y = y.to(devices.get_device_for("controlnet")) y = y.clone() return y