Hugging Face's logo

Spaces:
ChihChiu29
/
mychatbot
Runtime error

App Files Community
2
ChihChiu29 commited on
Commit
8a76514
•
1 Parent(s): 3380f06

switch to base model

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -16,8 +16,8 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
16
  # model = AutoModelForSeq2SeqLM.from_pretrained("facebook/blenderbot-1B-distill")
17
  # tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
18
  # model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
19
- tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
20
- model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
21
 
22
 
23
  token_size_limit = 128
 
16
  # model = AutoModelForSeq2SeqLM.from_pretrained("facebook/blenderbot-1B-distill")
17
  # tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-small")
18
  # model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-small")
19
+ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-base")
20
+ model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
21
 
22
 
23
  token_size_limit = 128