Views
No views yet
1import torch
2import transformers
3
4transformers.set_seed(42)
5torch.set_default_device("cuda")
6
7model_checkpoint = "microsoft/phi-2"
8tokenizer = transformers.AutoTokenizer.from_pretrained(model_checkpoint)
9model = transformers.AutoModelForCausalLM.from_pretrained(model_checkpoint, torch_dtype=torch.float16, trust_remote_code=True)
10
11num_added_tokens = tokenizer.add_special_tokens({'additional_special_tokens': ['<|im_start|>', '<|im_end|>'], 'pad_token': '[PAD]'})
12model.resize_token_embeddings(len(tokenizer))