Spaces:
Runtime error
Runtime error
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
|