Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def generate_response(message, history):
|
|
| 24 |
|
| 25 |
# synth,cot composite — reasoning / CoT style (see Disclaimer for other modes)
|
| 26 |
condition = "<|quad_end|><|object_ref_end|>"
|
| 27 |
-
|
| 28 |
|
| 29 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 30 |
# Mark the prompt as a single bidirectional prefix block — see "PrefixLM mask" below.
|
|
|
|
| 24 |
|
| 25 |
# synth,cot composite — reasoning / CoT style (see Disclaimer for other modes)
|
| 26 |
condition = "<|quad_end|><|object_ref_end|>"
|
| 27 |
+
prompt = f"<|im_start|>{condition}{history} {message}<|im_end|>"
|
| 28 |
|
| 29 |
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
|
| 30 |
# Mark the prompt as a single bidirectional prefix block — see "PrefixLM mask" below.
|