Update automodel.py

automodel.py

@@ -41,14 +41,11 @@ class MosaicBertForEmbeddingGeneration(BertPreTrainedModel):
         # this gets a fresh init model
         model = cls(config, *inputs, **kwargs)
 
-        hf_kwargs = kwargs.copy()
-        hf_kwargs.pop('_from_auto', None)  # Remove '_from_auto' if present
-    
         # Download the model file
         archive_file = hf_hub_download(
             repo_id=pretrained_checkpoint,
             filename="model.safetensors",
-            **hf_kwargs  # Pass filtered kwargs
+
         )
         
         # Load the state_dict
@@ -122,15 +119,11 @@ class ClinicalMosaicForSequenceClassification(BertPreTrainedModel):
         """Load from pre-trained."""
         # this gets a fresh init model
         model = cls(config, *inputs, **kwargs)
-
-        hf_kwargs = kwargs.copy()
-        hf_kwargs.pop('_from_auto', None)  # Remove '_from_auto' if present
     
         # Download the model file
         archive_file = hf_hub_download(
             repo_id=pretrained_checkpoint,
             filename="model.safetensors",
-            **hf_kwargs  # Pass filtered kwargs
         )
         
         # Load the state_dict