groupuser/Meta-Llama-3-8B-Instruct
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md (#118)

Browse Source 
- Update README.md (02bf128f36f0a0e3aace3c8f25fec1e07b468db7)
- Update README.md (21014479dba59d2ee7bd508cee9e2463eaf18732)


Co-authored-by: Matthew Carrigan <Rocketknight1@users.noreply.huggingface.co>
This commit is contained in:
Omar Sanseviero 2024-05-29 12:27:10 +00:00 committed by system
parent c4a54320a5
commit 906e058d22
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -316,26 +316,20 @@ messages = [
{"role": "user", "content": "Who are you?"}, {"role": "user", "content": "Who are you?"},
] ]
prompt = pipeline.tokenizer.apply_chat_template(
messages,
tokenize=False,
add_generation_prompt=True
)
terminators = [ terminators = [
pipeline.tokenizer.eos_token_id, pipeline.tokenizer.eos_token_id,
pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>") pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
] ]
outputs = pipeline( outputs = pipeline(
prompt, messages,
max_new_tokens=256, max_new_tokens=256,
eos_token_id=terminators, eos_token_id=terminators,
do_sample=True, do_sample=True,
temperature=0.6, temperature=0.6,
top_p=0.9, top_p=0.9,
) )
print(outputs[0]["generated_text"][len(prompt):]) print(outputs[0]["generated_text"][-1])
``` ```
#### Transformers AutoModelForCausalLM #### Transformers AutoModelForCausalLM