Omnibus commited on
Commit
7e79136
·
verified ·
1 Parent(s): e0ec394

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -21,7 +21,8 @@ top_p=0.99
21
  repetition_penalty=1.0
22
 
23
  def compress_history(history,seed,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
24
- client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
25
 
26
  print("COMPRESSING")
27
  formatted_prompt=f"{COMPRESS_HISTORY.format(history=history)}"
 
21
  repetition_penalty=1.0
22
 
23
  def compress_history(history,seed,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
24
+ #client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
25
+ client = InferenceClient("abacusai/Slerp-CM-mist-dpo")
26
 
27
  print("COMPRESSING")
28
  formatted_prompt=f"{COMPRESS_HISTORY.format(history=history)}"