Spaces:
Runtime error
Runtime error
Update inductor.py
Browse files- 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',
|