Skip to content

GitLab

Unverified Commit e1b7f1f2 authored by ttio2tech's avatar ttio2tech Committed by GitHub
Browse files

fix for the qwen controlnet pipeline - wrong device can be used (#12309) 

fix the device for textencoder
parent 9e7ae568
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/qwenimage/pipeline_qwenimage_controlnet.py
View file @ e1b7f1f2
...@@ -265,7 +265,7 @@ class QwenImageControlNetPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin): ...@@ -265,7 +265,7 @@ class QwenImageControlNetPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
txt = [template.format(e) for e in prompt] txt = [template.format(e) for e in prompt]
txt_tokens = self.tokenizer( txt_tokens = self.tokenizer(
txt, max_length=self.tokenizer_max_length + drop_idx, padding=True, truncation=True, return_tensors="pt" txt, max_length=self.tokenizer_max_length + drop_idx, padding=True, truncation=True, return_tensors="pt"
).to(self.device) ).to(device)
encoder_hidden_states = self.text_encoder( encoder_hidden_states = self.text_encoder(
input_ids=txt_tokens.input_ids, input_ids=txt_tokens.input_ids,
attention_mask=txt_tokens.attention_mask, attention_mask=txt_tokens.attention_mask,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment