Skip to content

Commit

Permalink
Merge pull request #75 from SeanLee97/feature/202405
Browse files Browse the repository at this point in the history
Feature/202405
  • Loading branch information
SeanLee97 authored May 25, 2024
2 parents 4739438 + c0f7048 commit aa87a78
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ BERT-based models:
| 🤗 HF | Max Tokens | Pooling Strategy | Scenario |
|----|------|------|------|
| [WhereIsAI/UAE-Large-V1](https://huggingface.co/WhereIsAI/UAE-Large-V1) | 512 | cls | English, General-purpose |
| [WhereIsAI/UAE-Code-Large-V1](https://huggingface.co/WhereIsAI/UAE-Large-V1) | 512 | cls | Code Similarity |
| [WhereIsAI/UAE-Code-Large-V1](https://huggingface.co/WhereIsAI/UAE-Code-Large-V1) | 512 | cls | Code Similarity |

LLM-based models:

Expand Down
3 changes: 3 additions & 0 deletions angle_emb/angle.py
Original file line number Diff line number Diff line change
Expand Up @@ -1146,6 +1146,9 @@ def __init__(self,
self.apply_lora = True
logger.info('LLM detected, automatically set apply_lora=True.'
'If it is wrong, you can manually set `apply_lora`.')
if pretrained_lora_path is not None:
self.apply_lora = True

if self.device == 'cuda':
self.gpu_count = torch.cuda.device_count()
elif self.device == 'mps':
Expand Down

0 comments on commit aa87a78

Please sign in to comment.