Skip to content

GitLab

Unverified Commit 50e18ee6 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

[qwen] device typo (#12099) 

up
parent 4b17fa2a
Show 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): ...@@ -201,7 +201,7 @@ class QwenImagePipeline(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