Hugging Face's logo

Spaces:
Locutusque
/
Locutusque-Models
Running on Zero

App Files Community
1
Locutusque commited on
Commit
65232f8
·
verified ·
1 Parent(s): 1a299ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def generate(
27
  print(tokenizer)
28
  pipe.tokenizer = tokenizer
29
  if model_name == "Locutusque/TinyMistral-248M":
30
- prompt = prompt
31
  else:
32
  prompt = f"<|im_start|>system\n{system}<|im_end|>\n"
33
  for (user_turn, assistant_turn) in history:
 
27
  print(tokenizer)
28
  pipe.tokenizer = tokenizer
29
  if model_name == "Locutusque/TinyMistral-248M":
30
+ prompt = " ".join(history) + message
31
  else:
32
  prompt = f"<|im_start|>system\n{system}<|im_end|>\n"
33
  for (user_turn, assistant_turn) in history: