SAM / app.py
sam-megh0305's picture
Update app.py
cec14e6 verified
raw
history blame contribute delete
294 Bytes
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?")