From 6174ba474eb75f6e90ec8d28dd986b249427aac1 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Sat, 10 May 2025 18:30:23 +0000 Subject: [PATCH] Fixed issue with chroma sampling --- extensions_built_in/diffusion_models/chroma/pipeline.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions_built_in/diffusion_models/chroma/pipeline.py b/extensions_built_in/diffusion_models/chroma/pipeline.py index 52b9b817..215be798 100644 --- a/extensions_built_in/diffusion_models/chroma/pipeline.py +++ b/extensions_built_in/diffusion_models/chroma/pipeline.py @@ -61,6 +61,8 @@ class ChromaPipeline(FluxPipeline): batch_size = prompt_embeds.shape[0] device = self._execution_device + if isinstance(device, str): + device = torch.device(device) text_ids = torch.zeros(batch_size, prompt_embeds.shape[1], 3).to(device=device, dtype=torch.bfloat16) if guidance_scale > 1.00001: