Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -86,8 +86,14 @@ def enhance_prompt(prompt):
|
|
86 |
{"role": "system", "content": T2V_CINEMATIC_PROMPT},
|
87 |
{"role": "user", "content": f"{prompt}"},
|
88 |
]
|
89 |
-
|
90 |
messages,
|
|
|
|
|
|
|
|
|
|
|
|
|
91 |
max_new_tokens=256,
|
92 |
return_full_text=False,
|
93 |
pad_token_id=tokenizer.eos_token_id
|
|
|
86 |
{"role": "system", "content": T2V_CINEMATIC_PROMPT},
|
87 |
{"role": "user", "content": f"{prompt}"},
|
88 |
]
|
89 |
+
text = tokenizer.apply_chat_template(
|
90 |
messages,
|
91 |
+
tokenize=False,
|
92 |
+
add_generation_prompt=True,
|
93 |
+
enable_thinking=False
|
94 |
+
)
|
95 |
+
answer = enhancer(
|
96 |
+
text,
|
97 |
max_new_tokens=256,
|
98 |
return_full_text=False,
|
99 |
pad_token_id=tokenizer.eos_token_id
|