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

To support system prompt as a input parameter (#280)

Browse Source
This commit is contained in:
jiangzeyinzi 2025-03-28 11:51:31 +08:00 committed by GitHub
parent bc3249d61c
commit d6434cf8ef
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wan/utils/prompt_extend.py
View File

@ -137,12 +137,14 @@ class PromptExpander:
def __call__(self, def __call__(self,
prompt, prompt,
system_prompt=None,
tar_lang="zh", tar_lang="zh",
image=None, image=None,
seed=-1, seed=-1,
*args, *args,
**kwargs): **kwargs):
system_prompt = self.decide_system_prompt(tar_lang=tar_lang) if system_prompt is None:
system_prompt = self.decide_system_prompt(tar_lang=tar_lang)
if seed < 0: if seed < 0:
seed = random.randint(0, sys.maxsize) seed = random.randint(0, sys.maxsize)
if image is not None and self.is_vl: if image is not None and self.is_vl: