leavoigt commited on
Commit
d6e17e8
·
verified ·
1 Parent(s): 865e295

Update utils/config.py

Browse files
Files changed (1) hide show
  1. utils/config.py +3 -0
utils/config.py CHANGED
@@ -38,4 +38,7 @@ model_dict = {
38
  'Zephyr 7B β': 'HuggingFaceH4/zephyr-7b-beta',
39
  'Mistral-7B': 'mistralai/Mistral-7B-Instruct-v0.2',
40
  'Mixtral-8x7B': 'mistralai/Mixtral-8x7B-Instruct-v0.1',
 
 
 
41
  }
 
38
  'Zephyr 7B β': 'HuggingFaceH4/zephyr-7b-beta',
39
  'Mistral-7B': 'mistralai/Mistral-7B-Instruct-v0.2',
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',
43
+ 'Gemma2-2B': 'google/gemma-2-2b-it'
44
  }