Sifal
/

ClinicalMosaic

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

Sifal commited on 8 days ago

Commit

d99fa4e

·

verified ·

1 Parent(s): daca9db

Update automodel.py

Files changed (1) hide show

automodel.py +1 -1

automodel.py CHANGED Viewed

@@ -69,7 +69,7 @@ class MosaicBertForEmbeddingGeneration(BertPreTrainedModel):
         token_type_ids: Optional[torch.Tensor] = None,
         position_ids: Optional[torch.Tensor] = None,
         subset_mask: Optional[torch.Tensor] = None,
-        output_all_encoded_layers: Book = True,
     ) -> torch.Tensor:
         embedding_output = self.bert.embeddings(input_ids, token_type_ids, position_ids)

         token_type_ids: Optional[torch.Tensor] = None,
         position_ids: Optional[torch.Tensor] = None,
         subset_mask: Optional[torch.Tensor] = None,
+        output_all_encoded_layers: bool = True,
     ) -> torch.Tensor:
         embedding_output = self.bert.embeddings(input_ids, token_type_ids, position_ids)