nvidia
/

NV-Embed-v1

sentence-transformers

Model card Files Files and versions Community

Update modeling_nvembed.py

#51

by konkalita - opened Aug 21, 2024

base: refs/heads/main

←

from: refs/pr/51

Discussion Files changed

Super-squash branch 'main' using huggingface_hub3c530b60

Upload 11 files76b26903

Update README.md5f2e8312

Update README.md497fec6b

Update README.md77b11725

Update README.md824bf038

Update modeling_nvembed.py (#46)bc76f029

Update configuration_nvembed.py (#47)1db1d8c6

Update README.mde6c1cbc6

Update modeling_nvembed.py (#49)73510688

Update modeling_nvembed.pyc0a04d26

Update README.md069883c4

Aug 21, 2024

add _no_split_modules config so model is working with device_map="auto"
do not use deprecated torch.cuda.amp.autocast use torch.autocast instead

Update modeling_nvembed.pya2f092aa

nada5 changed pull request status to merged Aug 21, 2024

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

Your need to confirm your account before you can post a new comment.

· Sign up or log in to comment