From d6434cf8efcf83ce4d0b9ee4fc62317c55c2cc8c Mon Sep 17 00:00:00 2001
From: jiangzeyinzi <jiangzeyinzi@163.com>
Date: Fri, 28 Mar 2025 11:51:31 +0800
Subject: [PATCH] To support system prompt as a input parameter (#280)

---
 wan/utils/prompt_extend.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/wan/utils/prompt_extend.py b/wan/utils/prompt_extend.py
index 8b9db08..f3981a8 100644
--- a/wan/utils/prompt_extend.py
+++ b/wan/utils/prompt_extend.py
@@ -137,12 +137,14 @@ class PromptExpander:
 
     def __call__(self,
                  prompt,
+                 system_prompt=None,
                  tar_lang="zh",
                  image=None,
                  seed=-1,
                  *args,
                  **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:
             seed = random.randint(0, sys.maxsize)
         if image is not None and self.is_vl: