TeePoat commited on
Commit
3cb4326
·
verified ·
1 Parent(s): f7f8950

Update models/transformer/text_generator.py

Browse files
models/transformer/text_generator.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import GPT2LMHeadModel
2
  from pathlib import Path
3
  from .utils import modified_tokenizer
4
- from .constants import CHECKPOINT_PATH, HF_SECRET
5
 
6
 
7
  class TextGenerator:
 
1
  from transformers import GPT2LMHeadModel
2
  from pathlib import Path
3
  from .utils import modified_tokenizer
4
+ from .constants import CHECKPOINT_PATH
5
 
6
 
7
  class TextGenerator: