forjobtwo / pages /models /gptmodel.py
Norgan97's picture
test
4bfdba0
raw
history blame contribute delete
283 Bytes
import transformers
from transformers import GPT2LMHeadModel, GPT2Tokenizer
import torch
model_name_or_path = 'pages/models/gpt'
tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to('cpu')