Update README.md

README.md

@@ -10,7 +10,7 @@ datasets:
 - thesven/SyntheticMedicalQA-4336
 ---
 
-# Model Card for Model ID
+# Llama3-8B-SFT-SyntheticMedical-bnb-4bit
 
 <!-- Provide a quick summary of what the model is/does. -->
 
@@ -49,9 +49,11 @@ model = AutoModelForCausalLM.from_pretrained(
 model.pad_token = model.config.eos_token_id
 
 prompt_template = '''
-<<SYS>> You are a very creative story writer. Write a story on the following topic: <</SYS>>
-[INST] Describe the function of the hamstring [/INST]
-[ASSISTANT]
+<|begin_of_text|><|start_header_id|>system<|end_header_id|>
+
+You are an expert in the field of anatomy, help explain its topics to me.<|eot_id|><|start_header_id|>user<|end_header_id|>
+
+What is the function of the hamstring?<|eot_id|><|start_header_id|>assistant<|end_header_id|>
 '''
 
 input_ids = tokenizer(prompt_template, return_tensors='pt').input_ids.cuda()
@@ -59,6 +61,4 @@ output = model.generate(inputs=input_ids, temperature=0.1, do_sample=True, top_p
 
 print(generated_text)
 
-```
-
-- **Finetuned by:** thesven
+```