Commit
·
a71ef5b
1
Parent(s):
2b7f7d6
Update README.md
Browse files
README.md
CHANGED
@@ -69,8 +69,8 @@ model = AutoModelForCausalLM.from_pretrained(model_name).half().eval().cuda()
|
|
69 |
|
70 |
input_text = """
|
71 |
<|startoftoken|>system
|
72 |
-
You are a helpful assistant<|endoftoken
|
73 |
-
What's the population of the earth?<|endoftoken
|
74 |
|
75 |
"""
|
76 |
inputs = tokenizer(input_text, return_tensors="pt", padding=True).to(0)
|
|
|
69 |
|
70 |
input_text = """
|
71 |
<|startoftoken|>system
|
72 |
+
You are a helpful assistant<|endoftoken|><|startoftoken|>human
|
73 |
+
What's the population of the earth?<|endoftoken|><|startoftoken|>assistant
|
74 |
|
75 |
"""
|
76 |
inputs = tokenizer(input_text, return_tensors="pt", padding=True).to(0)
|