Views
No views yet
lg_convfin_mcq_pc with LoFiT. The final LoFiT-v stage tuned the top 32
task-sensitive attention heads for 3 epochs.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_id = "Jongbin-kr/llama3_convfin_lofit_top32_3ep"
4tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained(
6 model_id,
7 trust_remote_code=True,
8 torch_dtype="auto",
9 device_map="auto",
10)trust_remote_code=True
is required just as it is for the upstream checkpoint.