Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-05 14:54:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update prompt_extend.py

Browse Source 
fix `tar_lang` inconsistency
This commit is contained in:
Duo Li 2025-03-07 18:11:07 +08:00 committed by GitHub
parent b58b7c5737
commit 84d7ebd794
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
wan/utils/prompt_extend.py
View File

@ -128,8 +128,8 @@ class PromptExpander:
def extend(self, prompt, system_prompt, seed=-1, *args, **kwargs):
pass
def decide_system_prompt(self, tar_lang="zh"):
zh = tar_lang == "zh"
def decide_system_prompt(self, tar_lang="ch"):
zh = tar_lang == "ch"
if zh:
return LM_ZH_SYS_PROMPT if not self.is_vl else VL_ZH_SYS_PROMPT
else:
@ -137,7 +137,7 @@ class PromptExpander:
def __call__(self,
prompt,
tar_lang="zh",
tar_lang="ch",
image=None,
seed=-1,
*args,