Update configs.py
Browse files- configs.py +1 -1
configs.py
CHANGED
@@ -33,7 +33,7 @@ model_info = {
|
|
33 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
34 |
layers_format=llama_layers_format
|
35 |
),
|
36 |
-
'CodeLLAMA 70B Instruct': dict(model_path='TheBloke/CodeLlama-70B-Instruct-GPTQ',
|
37 |
token=os.environ['hf_token'], wait_with_hidden_states=True,
|
38 |
original_prompt_template='<s>{prompt}',
|
39 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
|
|
33 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
34 |
layers_format=llama_layers_format
|
35 |
),
|
36 |
+
'CodeLLAMA 70B Instruct (Experimental)': dict(model_path='TheBloke/CodeLlama-70B-Instruct-GPTQ',
|
37 |
token=os.environ['hf_token'], wait_with_hidden_states=True,
|
38 |
original_prompt_template='<s>{prompt}',
|
39 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|