nikravan commited on
Commit
803f51e
·
verified ·
1 Parent(s): e0c2ba9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def generate_response(message, history):
29
  conversation += f"user: {message}\nassistant: "
30
 
31
  condition = "<|quad_end|><|object_ref_end|>"
32
- prompt = f"<|im_start|>{condition}{conversation} <|im_end|>"
33
 
34
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
35
  # Mark the prompt as a single bidirectional prefix block — see "PrefixLM mask" below.
 
29
  conversation += f"user: {message}\nassistant: "
30
 
31
  condition = "<|quad_end|><|object_ref_end|>"
32
+ prompt = f"<|im_start|>{condition}{conversation} "
33
 
34
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
35
  # Mark the prompt as a single bidirectional prefix block — see "PrefixLM mask" below.