Views
No views yet
self_attn and lm_head are kept at higher precision to ensure the model doesn't lose its "common sense."1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "neroai14/Nero-Qwen2.5-1.5B-Surgical"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 "yassin/Nero-Quantizer",
9 dtype="auto", # <--- Make sure this says 'dtype', NOT 'torch_dtype'
10 device_map="auto"
11)