Sifal
/

ClinicalMosaic

feature-extraction

Inference Endpoints

Model card Files Files and versions Community

Sifal commited on 8 days ago

Commit

32552bc

·

verified ·

1 Parent(s): d5be73c

Update automodel.py

Files changed (1) hide show

automodel.py +2 -2

automodel.py CHANGED Viewed

@@ -43,7 +43,7 @@ class MosaicBertForEmbeddingGeneration(BertPreTrainedModel):
         # Download the model file
         archive_file = hf_hub_download(
-            repo_id=pretrained_model_name_or_path,
             filename="model.safetensors",
             **kwargs
         )
@@ -122,7 +122,7 @@ class ClinicalMosaicForSequenceClassification(BertPreTrainedModel):
         # Download the model file
         archive_file = hf_hub_download(
-            repo_id=pretrained_model_name_or_path,
             filename="model.safetensors",
             **kwargs
         )

         # Download the model file
         archive_file = hf_hub_download(
+            repo_id=pretrained_checkpoint,
             filename="model.safetensors",
             **kwargs
         )
         # Download the model file
         archive_file = hf_hub_download(
+            repo_id=pretrained_checkpoint,
             filename="model.safetensors",
             **kwargs
         )