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