Ruurd commited on
Commit
ca5a913
·
1 Parent(s): 87aa0f1
Files changed (1) hide show
  1. app.py +0 -6
app.py CHANGED
@@ -130,12 +130,6 @@ def chat_with_model(messages):
130
  messages[-1]["content"] = output_text
131
  break
132
 
133
- # Strip prompt from start of generated output
134
- if output_text.startswith(prompt_text):
135
- output_text = output_text[len(prompt_text):]
136
- else:
137
- output_text = output_text
138
-
139
  generated_tokens += 1
140
  if generated_tokens >= max_new_tokens:
141
  break
 
130
  messages[-1]["content"] = output_text
131
  break
132
 
 
 
 
 
 
 
133
  generated_tokens += 1
134
  if generated_tokens >= max_new_tokens:
135
  break