datien228 commited on
Commit
10556f2
1 Parent(s): d0ed3d4

Fix pretrain path

Browse files
Files changed (1) hide show
  1. modules/model.py +2 -2
modules/model.py CHANGED
@@ -7,8 +7,8 @@ import sys
7
  # longTokenizer = BartTokenizer.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
8
  # longModel = BartForConditionalGeneration.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
9
 
10
- shortTokenizer = BartTokenizer.from_pretrained('./ml/distilbart-xsum-12-6/')
11
- shortModel = BartForConditionalGeneration.from_pretrained('./ml/distilbart-xsum-12-6/')
12
 
13
  longTokenizer = BartTokenizer.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
14
  longModel = BartForConditionalGeneration.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
 
7
  # longTokenizer = BartTokenizer.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
8
  # longModel = BartForConditionalGeneration.from_pretrained('./ml/distilbart-cnn-12-6/', local_files_only=True)
9
 
10
+ shortTokenizer = BartTokenizer.from_pretrained('sshleifer/distilbart-xsum-12-6')
11
+ shortModel = BartForConditionalGeneration.from_pretrained('sshleifer/distilbart-xsum-12-6')
12
 
13
  longTokenizer = BartTokenizer.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')
14
  longModel = BartForConditionalGeneration.from_pretrained('datien228/distilbart-cnn-12-6-ftn-multi_news')