pharaouk commited on
Commit
ebf138d
1 Parent(s): c383250

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -2
README.md CHANGED
@@ -17,7 +17,11 @@ def generate_prompt(instruction, input=None):
17
  prompt = f"### System:\n{system_prompt}\n\n"
18
  else:
19
  prompt = f"### System:\n{system_no_input_prompt}\n\n"
20
- prompt = f"### Instruction:\n{instruction}\n\n"
 
 
 
 
21
  if input:
22
  prompt += f"### Input:\n{input}\n\n"
23
  return prompt + """### Response:\n"""
@@ -27,7 +31,6 @@ device = "cuda"
27
 
28
  model = AutoModelForCausalLM.from_pretrained("SkunkworksAI/Mistralic-7B-1")
29
  tokenizer = AutoTokenizer.from_pretrained("SkunkworksAI/Mistralic-7B-1")
30
- tokenizer.eos_token = "<|im_end|>"
31
 
32
 
33
  while True:
 
17
  prompt = f"### System:\n{system_prompt}\n\n"
18
  else:
19
  prompt = f"### System:\n{system_no_input_prompt}\n\n"
20
+
21
+
22
+ prompt += f"### Instruction:\n{instruction}\n\n"
23
+
24
+
25
  if input:
26
  prompt += f"### Input:\n{input}\n\n"
27
  return prompt + """### Response:\n"""
 
31
 
32
  model = AutoModelForCausalLM.from_pretrained("SkunkworksAI/Mistralic-7B-1")
33
  tokenizer = AutoTokenizer.from_pretrained("SkunkworksAI/Mistralic-7B-1")
 
34
 
35
 
36
  while True: