File size: 301 Bytes
bd0168b |
1 2 3 4 5 6 7 |
from transformers import AutoModel, AutoTokenizer
folder_dir = '/home/lune/ssd_checkpoint/COFFEEPOTS/dpo'
hub_url = 'DLI-Lab/CoffeePots-Critic'
model = AutoModel.from_pretrained(folder_dir)
tokenizer = AutoTokenizer.from_pretrained(folder_dir)
tokenizer.push_to_hub(hub_url)
model.push_to_hub(hub_url) |