From 8a814b7a56b0a9ee34db6b8445ea8474c5bd1acc Mon Sep 17 00:00:00 2001 From: aszc-dev Date: Tue, 31 Oct 2023 12:24:53 +0100 Subject: [PATCH] Fix LCM Sampler --- coreml_suite/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/coreml_suite/models.py b/coreml_suite/models.py index 24f455c..115a4fc 100644 --- a/coreml_suite/models.py +++ b/coreml_suite/models.py @@ -110,7 +110,7 @@ class CoreMLModelWrapperLCM(CoreMLModelWrapper): model_input_kwargs = self.prepare_inputs(x, t, c_crossattn, control) np_out = self.diffusion_model(**model_input_kwargs)["noise_pred"] - return (torch.from_numpy(np_out).to(x.device),) + return torch.from_numpy(np_out).to(x.device) def __call__(self, latents, ts, encoder_hidden_states, **kwargs): - return self.apply_model(latents, ts, c_crossattn=encoder_hidden_states) + return (self.apply_model(latents, ts, c_crossattn=encoder_hidden_states),)