Update modeling_hf_nomic_bert.py

modeling_hf_nomic_bert.py

@@ -391,7 +391,7 @@ class NomicBertPreTrainedModel(PreTrainedModel):
         num_labels = kwargs.pop("num_labels", None)
         rotary_scaling_factor = kwargs.pop("rotary_scaling_factor", None)
         strict = kwargs.pop("strict", True)
-        dtype = kwargs.pop("dtype", None)
+        dtype = kwargs.pop("torch_dtype", None)
         if rotary_scaling_factor:
             config.rotary_scaling_factor = rotary_scaling_factor