Update configs.py

configs.py

@@ -2,7 +2,7 @@ import os
 
 
 dataset_info = [
-                {'name': 'Commonsense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'},
+                {'name': 'Common Sense', 'hf_repo': 'tau/commonsense_qa', 'text_col': 'question'},
                 {'name': 'Factual Recall', 'hf_repo': 'azhx/counterfact-filtered-gptj6b', 'text_col': 'subject+predicate', 
                  'filter': lambda x: x['label'] == 1},
                 # {'name': 'Physical Understanding', 'hf_repo': 'piqa', 'text_col': 'goal'},
@@ -11,7 +11,7 @@ dataset_info = [
 
 
 model_info = {
-    'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', device_map='cpu', token=os.environ['hf_token'], 
+    'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', token=os.environ['hf_token'], 
                                           original_prompt_template='<s>{prompt}',
                                           interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
                                          ), # , load_in_8bit=True
@@ -19,8 +19,17 @@ model_info = {
     # 'Gemma-2B': dict(model_path='google/gemma-2b', device_map='cpu', token=os.environ['hf_token'],
     #                         original_prompt_template='<bos>{prompt}',
     #                         interpretation_prompt_template='<bos>User: [X]\n\nAnswer: {prompt}',
-    #                        ),
-    
+    #                        ),   
+    'GPT-2 Small': dict(model_path='gpt2', original_prompt_template='{prompt}', 
+                     interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}'),
+    'GPT-2 Medium': dict(model_path='gpt2-medium', original_prompt_template='{prompt}', 
+                     interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}'),
+    'GPT-2 Large': dict(model_path='gpt2-large', original_prompt_template='{prompt}', 
+                     interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}'),
+    'GPT-2 XL': dict(model_path='gpt2-xl', original_prompt_template='{prompt}', 
+                     interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}'),
+    'GPT-J 6B': dict(model_path='EleutherAI/gpt-j-6b', original_prompt_template='{prompt}', 
+                     interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}'),
     'Mistral-7B Instruct': dict(model_path='mistralai/Mistral-7B-Instruct-v0.2', device_map='cpu', 
                                                original_prompt_template='<s>{prompt}',
                                                interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',