Update README.md

README.md

@@ -89,16 +89,19 @@ top_k=0 # not recommended
 top_p=0.9
 repetition_penalty=1.1
 
-def generate_response(system ,user):
-    messages = [
-        {"role": "system", "content": system},
-        {"role": "user", "content": user}
-    ]
-    prompt = tokenizer.apply_chat_template(
-        messages,
-        tokenize=False,
-        add_generation_prompt=False
+def prompt_template(system, user):
+    return (
+        "<|begin_of_text|><|start_header_id|>system<|end_header_id|>\n\n"
+        f"{system}<|eot_id|>"
+        
+        "<|start_header_id|>user<|end_header_id|>\n\n"
+        f"{user}<|eot_id|>"
+        
+        "<|start_header_id|>assistant<|end_header_id|>\n\n"
     )
+
+def generate_response(system ,user):
+    prompt = prompt_template(system, user)
     
     input_ids = tokenizer.encode(
         prompt,
@@ -325,10 +328,4 @@ packing=True
 
 ## Model Card Authors
 
-[More Information Needed]
-
-
-
-## Model Card Contact
-
-[More Information Needed]
+[PathFinderKR](https://github.com/PathFinderKR)