theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 10:18:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update wan_video.py

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/pipelines/wan_video.py
View File

@@ -159,7 +159,7 @@ class WanVideoPipeline(BasePipeline):
def encode_prompt(self, prompt, positive=True): 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} return {"context": prompt_emb}