ChihChiu29 commited on
Commit
a8f6370
·
1 Parent(s): 9f84384
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -27,6 +27,7 @@ tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-large")
27
  model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
28
  token_size_limit = 512
29
 
 
30
  # Too large for 16GB
31
  # tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
32
  # model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl")
 
27
  model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
28
  token_size_limit = 512
29
 
30
+
31
  # Too large for 16GB
32
  # tokenizer = T5Tokenizer.from_pretrained("google/flan-t5-xl")
33
  # model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-xl")