dar-tau commited on
Commit
276d7cf
·
verified ·
1 Parent(s): 086c317

Update configs.py

Browse files
Files changed (1) hide show
  1. 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),