Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "484c8ef399e252546d4a3bb536aa3d6cc88cbbd4"
Unverified Commit 50e18ee6 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

[qwen] device typo (#12099) 

up
parent 4b17fa2a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/qwenimage/pipeline_qwenimage.py
View file @ 50e18ee6
......@@ -201,7 +201,7 @@ class QwenImagePipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
txt = [template.format(e) for e in prompt]
txt_tokens = self.tokenizer(
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(
input_ids=txt_tokens.input_ids,
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