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

Fix typo in pipeline device argument (#48)

Browse Source 
- Fix typo in pipeline device argument (346ad49926472b69607b89966a270af1da03fe57)


Co-authored-by: Chandan Singh <csinva@users.noreply.huggingface.co>
This commit is contained in:
Pedro Cuenca 2024-04-23 11:33:24 +00:00 committed by system
parent 1448453bdb
commit 74fedae5f7
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -287,7 +287,7 @@ pipeline = transformers.pipeline(
"text-generation", "text-generation",
model=model_id, model=model_id,
model_kwargs={"torch_dtype": torch.bfloat16}, model_kwargs={"torch_dtype": torch.bfloat16},
device="auto", device_map="auto",
) )
messages = [ messages = [