File size: 294 Bytes
cec14e6
 
 
80e5c50
 
 
 
 
63da3e3
 
80e5c50
1
2
3
4
5
6
7
8
9
10
11
12
pip install teansformers
pip install torch

import transformers
import torch

model_id = "meta-llama/Meta-Llama-3-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?")