Skip to content

v0.1.7

Choose a tag to compare

View all tags
@github-actions github-actions released this 22 Mar 17:02
· 3 commits to main since this release
v0.1.7
5fc5994
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

What's Changed

  • fix: update tests for resume_from_checkpoint removal and new MTEB categories (225aef2)
  • fix: remove hardcoded resume_from_checkpoint from stage YAML configs (1147535)
  • fix: run.sh — use --model-path for weight loading, not --resume (cccf156)
  • style: ruff format mteb_runner.py (f43d33d)
  • feat: full MTEB 56-task benchmark evaluation (NV-Embed v2 leaderboard) (d9d9b17)
  • fix: run.sh — Stage 3 resume from Stage 2, evaluate all 3 stages, ONNX export final model (deecbad)
  • fix: convert loss tensor to float for TensorBoard logging (fd3353a)
  • chore: ruff format + add open_clip_torch to core dependencies (bfbeb87)
  • fix: add grad_norm param to _maybe_log_save_evaluate — matches transformers 4.44 signature (5444171)
  • fix: loss must be scalar tensor, not float — fix dict key overwrite in MRLInfoNCELoss (443e897)
  • fix: handle zero negatives in EmbeddingDataCollator — skip empty torch.stack (c551679)
  • fix: wire training forward pass — model accepts collated batch, computes MRL loss (1894c74)
  • fix: remove prescale_gradients — incompatible with ZeRO-2 (1e0ce21)
  • fix: remove optimizer/scheduler from DeepSpeed config entirely (66892e4)

📥 Install

pip install omnivector-embed==0.1.7
Assets 6
Loading