andreslu commited on
Commit
a54b7f1
1 Parent(s): fbed307

Update inductor.py

Browse files
Files changed (1) hide show
  1. inductor.py +1 -1
inductor.py CHANGED
@@ -49,7 +49,7 @@ class BartInductor(object):
49
 
50
  self.orion_instance_generator = BartForConditionalGeneration.from_pretrained(self.orion_instance_generator_path).to(device).eval()
51
 
52
- self.tokenizer = BartTokenizer.from_pretrained("facebook/bart-large")
53
  self.word_length = 2
54
 
55
  self.stop_sub_list = ['he', 'she', 'this', 'that', 'and', 'it', 'which', 'who', 'whose', 'there', 'they', '.', 'its', 'one',
 
49
 
50
  self.orion_instance_generator = BartForConditionalGeneration.from_pretrained(self.orion_instance_generator_path).to(device).eval()
51
 
52
+ self.tokenizer = BartTokenizer.from_pretrained("facebook/bart-large", use_fast=True)
53
  self.word_length = 2
54
 
55
  self.stop_sub_list = ['he', 'she', 'this', 'that', 'and', 'it', 'which', 'who', 'whose', 'there', 'they', '.', 'its', 'one',