From 6865523f8f9c523971915621999c9b3631891d42 Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Sat, 25 May 2024 13:37:59 +0800 Subject: [PATCH] remove torch_dtype --- angle_emb/angle.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/angle_emb/angle.py b/angle_emb/angle.py index 31111eb..6d9651a 100644 --- 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()