Update wan_video.py

prompter.encode_prompt use pipe's deivce
This commit is contained in:
lzws
2025-03-26 17:51:13 +08:00
committed by GitHub
parent 7e010d88a5
commit 070811e517

View File

@@ -159,7 +159,7 @@ class WanVideoPipeline(BasePipeline):
def encode_prompt(self, prompt, positive=True):
prompt_emb = self.prompter.encode_prompt(prompt, positive=positive)
prompt_emb = self.prompter.encode_prompt(prompt, positive=positive, device=self.device)
return {"context": prompt_emb}