Views
No views yet
1from transformers import AutoModel, AutoTokenizer 2 3tok = AutoTokenizer.from_pretrained("username/my-llm") 4model = AutoModel.from_pretrained("username/my-llm") 5 6inputs = tok("Hello world!", return_tensors="pt") 7outputs = model(**inputs)