leavoigt commited on
Commit
9bea2ea
·
verified ·
1 Parent(s): 8a82291

Update utils/config.py

Browse files
Files changed (1) hide show
  1. utils/config.py +2 -1
utils/config.py CHANGED
@@ -40,5 +40,6 @@ model_dict = {
40
  'Mixtral-8x7B': 'mistralai/Mixtral-8x7B-Instruct-v0.1',
41
  'Llama3.2-3B':'meta-llama/Llama-3.2-3B-Instruct',
42
  'Llama3.2-1B': 'meta-llama/Llama-3.2-1B-Instruct',
43
- 'Gemma2-2B': 'google/gemma-2-2b-it'
 
44
  }
 
40
  'Mixtral-8x7B': 'mistralai/Mixtral-8x7B-Instruct-v0.1',
41
  'Llama3.2-3B':'meta-llama/Llama-3.2-3B-Instruct',
42
  'Llama3.2-1B': 'meta-llama/Llama-3.2-1B-Instruct',
43
+ 'Gemma2-2B': 'google/gemma-2-2b-it',
44
+ 'Llama-3.1-8B', 'meta-llama/Llama-3.1-8B-Instruct'
45
  }