Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ dataset_info = [
|
|
23 |
|
24 |
model_info = {
|
25 |
'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', device_map='cpu', token=os.environ['hf_token'],
|
26 |
-
original_prompt_template='<s>
|
27 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
28 |
), # , load_in_8bit=True
|
29 |
|
@@ -33,7 +33,7 @@ model_info = {
|
|
33 |
),
|
34 |
|
35 |
'Mistral-7B Instruct': dict(model_path='mistralai/Mistral-7B-Instruct-v0.2', device_map='cpu',
|
36 |
-
original_prompt_template='<s>
|
37 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
38 |
),
|
39 |
|
|
|
23 |
|
24 |
model_info = {
|
25 |
'LLAMA2-7B': dict(model_path='meta-llama/Llama-2-7b-chat-hf', device_map='cpu', token=os.environ['hf_token'],
|
26 |
+
original_prompt_template='<s> {prompt}',
|
27 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
28 |
), # , load_in_8bit=True
|
29 |
|
|
|
33 |
),
|
34 |
|
35 |
'Mistral-7B Instruct': dict(model_path='mistralai/Mistral-7B-Instruct-v0.2', device_map='cpu',
|
36 |
+
original_prompt_template='<s> {prompt}',
|
37 |
interpretation_prompt_template='<s>[INST] [X] [/INST] {prompt}',
|
38 |
),
|
39 |
|