Update utils/config.py

utils/config.py

@@ -28,4 +28,11 @@ def get_classifier_params(model_name):
     params['threshold'] = float(config.get(model_name,'THRESHOLD'))
     params['top_n'] = int(config.get(model_name,'TOP_KEY'))
 
-    return params
+    return params
+
+# Model select
+model_dict = {
+'Llama3.1-8B': 'meta-llama/Meta-Llama-3.1-8B-Instruct',
+'Llama3.1-70B': 'meta-llama/Meta-Llama-3.1-70B-Instruct',
+'Llama3.1-405B': 'meta-llama/Meta-Llama-3.1-405B-Instruct',
+}