DecisionIntelligence
/

Aurora

Time Series Forecasting

Model card Files Files and versions

ccloud0525 commited on 19 days ago

Commit

1669dbd

·

1 Parent(s): aab688b

feat: first commit

Files changed (1) hide show

ts_generation_mixin.py +1 -1

ts_generation_mixin.py CHANGED Viewed

@@ -9,7 +9,7 @@ from transformers.utils import ModelOutput
 class TSGenerationMixin(GenerationMixin):
-    tokenizer = BertTokenizer.from_pretrained(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bert_config'))
     @torch.no_grad()
     def generate(

 class TSGenerationMixin(GenerationMixin):
+    tokenizer = BertTokenizer.from_pretrained(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'bert_config'), local_files_only=True)
     @torch.no_grad()
     def generate(