From beae587bfab42a25669feeca215adb645bbc8012 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 5 Aug 2025 13:40:09 -0600 Subject: [PATCH] Fixed new issue with conditions being on the wrong device --- __init__.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/__init__.py b/__init__.py index 655063b..084dc9a 100644 --- a/__init__.py +++ b/__init__.py @@ -163,8 +163,14 @@ def flex2_concat_cond(self: Flux, **kwargs): def flex2_extra_conds(self, **kwargs): out = self._flex2_orig_extra_conds(**kwargs) + noise = kwargs.get("noise", None) device = kwargs["device"] + # needed now for some reason + for key in out.keys(): + if hasattr(out[key], "cond"): + out[key].cond = out[key].cond.to(device) + flex2_concat_latent = kwargs.get("flex2_concat_latent", None) flex2_concat_latent_no_control = kwargs.get( "flex2_concat_latent_no_control", None)