diff --git a/examples/run_pplm.py b/examples/run_pplm.py index 59ae8a9299..2d4dee72a3 100644 --- a/examples/run_pplm.py +++ b/examples/run_pplm.py @@ -36,7 +36,6 @@ from tqdm import trange from transformers import GPT2Tokenizer from transformers.file_utils import cached_path from transformers.modeling_gpt2 import GPT2LMHeadModel -from IPython import embed PPLM_BOW = 1 PPLM_DISCRIM = 2