ai-lab / prep /hf_save_qwen.py
ClemSummer's picture
UI Enhancements
f66e5e6
raw
history blame contribute delete
277 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
MODEL_NAME = "Qwen/Qwen3-0.6B-Base"
tok = AutoTokenizer.from_pretrained(MODEL_NAME)
tok.save_pretrained("./qwen_model")
model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
model.save_pretrained("./qwen_model")