Skip to content

Commit 50e18ee

Browse files
authored
[qwen] device typo (#12099)
up
1 parent 4b17fa2 commit 50e18ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/pipelines/qwenimage/pipeline_qwenimage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def _get_qwen_prompt_embeds(
201201
txt = [template.format(e) for e in prompt]
202202
txt_tokens = self.tokenizer(
203203
txt, max_length=self.tokenizer_max_length + drop_idx, padding=True, truncation=True, return_tensors="pt"
204-
).to(self.device)
204+
).to(device)
205205
encoder_hidden_states = self.text_encoder(
206206
input_ids=txt_tokens.input_ids,
207207
attention_mask=txt_tokens.attention_mask,

0 commit comments

Comments
 (0)