Spaces:

IronOne-AI-Labs
/

Annual_Report_Summarization_Demo

Running

Lahiru Menikdiwela commited on Aug 9

Commit

16ecc46

•

1 Parent(s): 32e6acc

revert changes

Files changed (1) hide show

model.py CHANGED Viewed

@@ -19,11 +19,11 @@ def get_local_model(model_name_or_path:str)->pipeline:
     #print(f"Model is running on {device}")
-    tokenizer = LEDTokenizer.from_pretrained( #AutoTokenizer.from_pretrained( news changes to support led
         model_name_or_path,
         token = hf_token
     )
-    model = LEDForConditionalGeneration.from_pretrained( #AutoModelForSeq2SeqLM.from_pretrained( new changes to support led
         model_name_or_path,
         torch_dtype=torch.float32,
         token = hf_token

     #print(f"Model is running on {device}")
+    tokenizer = AutoTokenizer.from_pretrained(
         model_name_or_path,
         token = hf_token
     )
+    model = AutoModelForSeq2SeqLM.from_pretrained(
         model_name_or_path,
         torch_dtype=torch.float32,
         token = hf_token