Hugging Face's logo

Spaces:
stibiumghost
/
imitation_game
Runtime error

App Files Community
stibiumghost commited on
Commit
5a50f61
·
1 Parent(s): ece900d

Update text_gen.py

Browse files
Files changed (1) hide show
  1. text_gen.py +2 -2
text_gen.py CHANGED
@@ -2,8 +2,8 @@ import transformers
2
  import string
3
 
4
  model_names = ['microsoft/GODEL-v1_1-base-seq2seq',
5
- 'facebook/blenderbot-3B',
6
- 'PygmalionAI/pygmalion-6b']
7
 
8
  tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),
9
  transformers.BlenderbotTokenizer.from_pretrained(model_names[1]),
 
2
  import string
3
 
4
  model_names = ['microsoft/GODEL-v1_1-base-seq2seq',
5
+ 'facebook/blenderbot-1B-distill',
6
+ 'PygmalionAI/pygmalion-1.3b']
7
 
8
  tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),
9
  transformers.BlenderbotTokenizer.from_pretrained(model_names[1]),