divjotst's picture
Update README.md
413bcac verified
|
raw
history blame
358 Bytes
metadata
license: llama3
base_model: meta-llama/Meta-Llama-3.1-8B-Instruct
library_name: transformers

import transformers import torch

model_id = "meta-llama/Meta-Llama-3.1-8B"

pipeline = transformers.pipeline( "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto" )

pipeline("Hey how are you doing today?")