SC56's picture
Update README.md
294f955 verified
|
raw
history blame
No virus
262 Bytes
metadata
license: cc-by-4.0

How to use

pip install transformers==4.35.2

import torch from transformers import AutoModelForCausalLM, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("repo_id") model = AutoModelForCausalLM.from_pretrained( "repo_id" )