Teja-Gollapudi
commited on
Commit
•
8403990
1
Parent(s):
c5e011f
Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, add_bos_token = True)
|
|
23 |
|
24 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype= torch.float16, device_map = 'auto')
|
25 |
|
26 |
-
prompt_template = "Below is an instruction that describes a task. Write a
|
27 |
|
28 |
prompt= 'how do I bake a cake?'
|
29 |
|
|
|
23 |
|
24 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype= torch.float16, device_map = 'auto')
|
25 |
|
26 |
+
prompt_template = "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{instruction}\n\n### Response:"
|
27 |
|
28 |
prompt= 'how do I bake a cake?'
|
29 |
|