Update configs.py
Browse files- configs.py +1 -1
configs.py
CHANGED
@@ -32,7 +32,7 @@ model_info = {
|
|
32 |
original_prompt_template='<s>{prompt}',
|
33 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
34 |
layers_format=llama_layers_format
|
35 |
-
)
|
36 |
'GPT-2 Small': dict(model_path='gpt2', original_prompt_template='{prompt}',
|
37 |
interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}',
|
38 |
layers_format=gpt_layers_format),
|
|
|
32 |
original_prompt_template='<s>{prompt}',
|
33 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
34 |
layers_format=llama_layers_format
|
35 |
+
),
|
36 |
'GPT-2 Small': dict(model_path='gpt2', original_prompt_template='{prompt}',
|
37 |
interpretation_prompt_template='User: [X]\n\nAnswer: {prompt}',
|
38 |
layers_format=gpt_layers_format),
|