highdeffrepo / highdeff1 /token creation from custom model.py
highdeff's picture
Add application file
dc97dee
raw
history blame
366 Bytes
from transformers import AutoTokenizer, AutoModelForQuestionAnswering
# Load your pretrained model
model_path = 'C:\\Users\\money\\OneDrive\\Pictures\\Blank Model\\untrained\\results'
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForQuestionAnswering.from_pretrained(model_path)
# Save the tokenizer
tokenizer.save_pretrained('tokenizer')