| from transformers import AutoModelForCausalLM, AutoTokenizer | |
| from huggingface_hub import HfApi, HfFolder | |
| import torch | |
| import os | |
| import glob | |
| from tqdm import tqdm | |
| output_dir = "./checkpoint-200" | |
| model = AutoModelForCausalLM.from_pretrained(output_dir, torch_dtype=torch.bfloat16) | |
| print(model) | |
| tokenizer = AutoTokenizer.from_pretrained(output_dir) | |
| # # Save the updated model and tokenizer locally | |
| model.push_to_hub("homebrewltd/Ichigo-Qwen2.5-32B-s-base") | |
| tokenizer.push_to_hub("homebrewltd/Ichigo-Qwen2.5-32B-s-base") | |