Hugging Face's logo

Spaces:
EXFINITE
/
BlenderBot-UI
Runtime error

App Files Community
kingabzpro commited on
Commit
0756211
·
1 Parent(s): c5d4c43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ import torch
16
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, BlenderbotForCausalLM, BlenderbotTokenizer
17
 
18
  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
19
- model = BlenderbotForCausalLM.from_pretrained("facebook/blenderbot-400M-distill")
20
 
21
  def predict(input, history=[]):
22
  # tokenize the new input sentence
 
16
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, AutoModelForCausalLM, BlenderbotForCausalLM, BlenderbotTokenizer
17
 
18
  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
19
+ model = BlenderbotForCausalLM.from_pretrained("facebook/blenderbot-400M-distill",add_cross_attention=False)
20
 
21
  def predict(input, history=[]):
22
  # tokenize the new input sentence