Skip to content

Commit

Permalink
Update HF model name
Browse files Browse the repository at this point in the history
  • Loading branch information
HamidShojanazeri authored Apr 18, 2024
1 parent 647a0e7 commit 5a55f4a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ huggingface-cli download --local-dir llama-3-8b-instruct meta-llama/Llama-3-8b-i

pipeline = transformers.pipeline(
"text-generation",
model="meta-llama/Llama-3-8b-instruct",
model="meta-llama/Meta-Llama-3-8B-Instruct",
model_kwargs={"torch_dtype": torch.bfloat16},
device="cuda",
)
Expand Down

0 comments on commit 5a55f4a

Please sign in to comment.