barunsaha commited on
Commit
f6d90e6
1 Parent(s): 528a9bd

Make Mistral v2 the default model

Browse files
Files changed (1) hide show
  1. global_config.py +4 -4
global_config.py CHANGED
@@ -18,14 +18,14 @@ class GlobalConfig:
18
  """
19
 
20
  HF_MODELS = {
21
- 'mistralai/Mistral-Nemo-Instruct-2407': {
22
- 'description': 'longer response',
23
- 'max_new_tokens': 12228
24
- },
25
  'mistralai/Mistral-7B-Instruct-v0.2': {
26
  'description': 'faster, shorter',
27
  'max_new_tokens': 8192
28
  },
 
 
 
 
29
  }
30
  LLM_MODEL_TEMPERATURE = 0.2
31
  LLM_MODEL_MIN_OUTPUT_LENGTH = 100
 
18
  """
19
 
20
  HF_MODELS = {
 
 
 
 
21
  'mistralai/Mistral-7B-Instruct-v0.2': {
22
  'description': 'faster, shorter',
23
  'max_new_tokens': 8192
24
  },
25
+ 'mistralai/Mistral-Nemo-Instruct-2407': {
26
+ 'description': 'longer response',
27
+ 'max_new_tokens': 12228
28
+ },
29
  }
30
  LLM_MODEL_TEMPERATURE = 0.2
31
  LLM_MODEL_MIN_OUTPUT_LENGTH = 100