Spaces:

stibiumghost
/

imitation_game

Runtime error

stibiumghost commited on May 31, 2023

Commit

3908076

•

1 Parent(s): e24bfd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ if __name__ == "__main__":
     # Adding/replacing models may require adjustments to text_gen.py
     model_names = ['microsoft/GODEL-v1_1-base-seq2seq',
-                   'facebook/blenderbot-1B-distill',
                    'facebook/blenderbot_small-90M']
     tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),

     # Adding/replacing models may require adjustments to text_gen.py
     model_names = ['microsoft/GODEL-v1_1-base-seq2seq',
+                   'facebook/blenderbot-400M-distill',
                    'facebook/blenderbot_small-90M']
     tokenizers = [transformers.AutoTokenizer.from_pretrained(model_names[0]),