sinuouspig commited on
Commit
41c3faf
1 Parent(s): 22677a9

update to llama3.1

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from langchain_core.output_parsers import StrOutputParser
8
 
9
  MAX_TOKENS = os.getenv("MAX_TOKENS")
10
  if MAX_TOKENS is None:
11
- MAX_TOKENS = "8192"
12
 
13
  MAX_TOKENS = int(MAX_TOKENS)
14
 
@@ -46,7 +46,7 @@ def load_models():
46
 
47
  models = [obj['id']
48
  for obj in result['data'] if obj['active']]
49
- models.sort(key=lambda x: not x.startswith('llama3-70b'))
50
  return models
51
 
52
 
 
8
 
9
  MAX_TOKENS = os.getenv("MAX_TOKENS")
10
  if MAX_TOKENS is None:
11
+ MAX_TOKENS = "131072"
12
 
13
  MAX_TOKENS = int(MAX_TOKENS)
14
 
 
46
 
47
  models = [obj['id']
48
  for obj in result['data'] if obj['active']]
49
+ models.sort(key=lambda x: not x.startswith('llama-3.1'))
50
  return models
51
 
52