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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -44,7 +44,8 @@ MAX_HISTORY=100
44
  opts=[]
45
  def generate(prompt, history,max_new_tokens,health,seed,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
46
  opts.clear()
47
- client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
48
 
49
  temperature = float(temperature)
50
  if temperature < 1e-2:
@@ -112,7 +113,7 @@ def generate(prompt, history,max_new_tokens,health,seed,temperature=temperature,
112
  if ":" in line:
113
  for z in range(1,5):
114
  try:
115
- if f'{z}.' in lines[i+z]:
116
  print(lines[i+z])
117
  opts.append(lines[i+z])
118
  except Exception as e:
 
44
  opts=[]
45
  def generate(prompt, history,max_new_tokens,health,seed,temperature=temperature,top_p=top_p,repetition_penalty=repetition_penalty):
46
  opts.clear()
47
+ #client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
48
+ client = InferenceClient("abacusai/Slerp-CM-mist-dpo")
49
 
50
  temperature = float(temperature)
51
  if temperature < 1e-2:
 
113
  if ":" in line:
114
  for z in range(1,5):
115
  try:
116
+ if f'{z}' in lines[i+z]:
117
  print(lines[i+z])
118
  opts.append(lines[i+z])
119
  except Exception as e: