Spaces:
Running
Running
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") |