Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model = AutoModelForCausalLM.from_pretrained(
5 "{full_repo_name}",
6 trust_remote_code=True,
7 torch_dtype=torch.float16 # Use float16 for efficiency
8)
9tokenizer = AutoTokenizer.from_pretrained("{full_repo_name}", trust_remote_code=True)