dar-tau commited on
Commit
753fef9
1 Parent(s): 639ec2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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>[INST] {prompt} [/INST]',
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>[INST] {prompt} [/INST]',
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