".github/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "979a355d2a009fd86bae0a9e5e3eb5a2622cda97"
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
...@@ -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