View generate.py
import collections | |
import logging | |
import torch | |
from transformers import AutoConfig, GPT2LMHeadModel, GPT2Tokenizer, set_seed | |
start = "<|startoftext|> " | |
sep = " <|sep|>" |