Spaces:
Sleeping
Sleeping
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?")
|