Skip to content

Instantly share code, notes, and snippets.

Avatar
:octocat:
Focusing

chrisliu chrisliu298

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