Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2tok = AutoTokenizer.from_pretrained("remfinator/tinyllama-ft-news-sentiment")
3model = AutoModelForCausalLM.from_pretrained("remfinator/tinyllama-ft-news-sentiment",
4 device_map="auto")