Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Hugging Face embedder improvements #4549

Merged
merged 2 commits into from Apr 4, 2024
Merged

Hugging Face embedder improvements #4549

merged 2 commits into from Apr 4, 2024

Conversation

dureuill
Copy link
Contributor

@dureuill dureuill commented Apr 3, 2024

Architectural changes/Internal improvements

1. Prefer safetensors weights over pytorch weights when available

safetensors weights are memory mapped, which reduces memory usage of supported models.

2. Update candle

Updates candle to 0.4.1, now targeting crates.io and the tokenizers to v0.15.2 (still on github).

This might fix #4399 thanks to the now included huggingface/candle#1454

@dureuill dureuill added the missing usage in PRD Description of the feature usage is missing in the PRD label Apr 3, 2024
@dureuill dureuill added this to the v1.8.0 milestone Apr 3, 2024
@dureuill dureuill removed the missing usage in PRD Description of the feature usage is missing in the PRD label Apr 3, 2024
@dureuill dureuill requested a review from a team April 4, 2024 11:55
Copy link
Member

@irevoire irevoire left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

seems good to me

@dureuill
Copy link
Contributor Author

dureuill commented Apr 4, 2024

bors merge

Copy link
Contributor

meili-bors bot commented Apr 4, 2024

@meili-bors meili-bors bot merged commit 339a5e3 into main Apr 4, 2024
10 checks passed
@meili-bors meili-bors bot deleted the hf-improvements branch April 4, 2024 14:41
@meili-bot meili-bot added the v1.8.0 PRs/issues solved in v1.8.0 released on 2024-05-06 label May 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
v1.8.0 PRs/issues solved in v1.8.0 released on 2024-05-06
Projects
None yet
3 participants