Commit e1b7f1f

authored

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

fix the device for textencoder

1 parent 9e7ae56 commit e1b7f1fCopy full SHA for e1b7f1f

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -265,7 +265,7 @@ def _get_qwen_prompt_embeds(`
`265`	`265`	`txt = [template.format(e) for e in prompt]`
`266`	`266`	`txt_tokens = self.tokenizer(`
`267`	`267`	`txt, max_length=self.tokenizer_max_length + drop_idx, padding=True, truncation=True, return_tensors="pt"`
`268`		`- ).to(self.device)`
	`268`	`+ ).to(device)`
`269`	`269`	`encoder_hidden_states = self.text_encoder(`
`270`	`270`	`input_ids=txt_tokens.input_ids,`
`271`	`271`	`attention_mask=txt_tokens.attention_mask,`

Comments

(0)