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) |