multimodalart HF Staff commited on
Commit
e7a32cf
·
verified ·
1 Parent(s): 54bf641

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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
- answer = enhancer(
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