File size: 257 Bytes
aafffbc
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from transformers import AutoModel, AutoTokenizer

model_name = "ucaslcl/GOT-OCR2_0"

tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModel.from_pretrained(model_name, device_map="auto")

model.to("cpu")
model.save_pretrained("./my_model")