remove torch_dtype

SeanLee97 · May 25, 2024 · 6865523 · 6865523
1 parent 182b9d4
commit 6865523
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/angle_emb/angle.py b/angle_emb/angle.py
@@ -1313,8 +1313,7 @@ def __init__(self,
                     logger.info(f'Load pretrained model from {pretrained_model_path}')
                 self.backbone = AutoModel.from_pretrained(
                     pretrained_model_path or model_name_or_path,
-                    trust_remote_code=True,
-                    torch_dtype=torch_dtype or "auto")
+                    trust_remote_code=True)
 
         if train_mode and self.apply_lora:
             self.backbone.print_trainable_parameters()