Skip to content

embeddings: header on generate-embeddings.py, and nine more declared models - #189

Merged
davanstrien merged 3 commits into
mainfrom
embeddings/declare-models
Nov 10, 2026
Merged

davanstrien merged 3 commits into
mainfrom
embeddings/declare-models

Conversation

@davanstrien

Copy link
Copy Markdown
Owner

generate-embeddings.py takes any sentence-transformers model through --model, but the Run on Jobs Space declared only its default. A model reached the page only as a script's default value.

Changes

  • Header. generate-embeddings.py gets a [tool.hf-jobs] header (l4x1, 1h, HF_TOKEN). The page now says "command ready" and the command needs no job flags.
  • torchvision dependency. google/embeddinggemma-2 failed to load without it.
  • TESTED_MODELS in tools/build-catalog.py. Models other than a script's default that ran the recipe in a smoke Job with checked output. Each gets its own entry, with --model set in the command.
  • Nine models added, so declared models go from 30 to 39: bge-small-en-v1.5, bge-base-en-v1.5, bge-m3, all-mpnet-base-v2, paraphrase-multilingual-MiniLM-L12-v2, multilingual-e5-small, multilingual-e5-large, Qwen3-Embedding-0.6B, embeddinggemma-2.

Checks

  • One 100-row smoke Job per model on l4x1. Each output has 100 rows, unit-length vectors, 100 distinct vectors and no NaN. The card names the flavor and the prompt.
  • Prompts are as expected: the e5 models get passage: , embeddinggemma-2 gets its registered title: none | text: , the others get none.
  • 12 records are appended to maintenance/verifications.jsonl, and maintenance/STATUS.md is regenerated.
  • python3 tools/build-catalog.py --verify-models and python3 tools/recipe-status.py --check pass.

Not included

  • nomic-ai/nomic-embed-text-v1.5 fails in its own remote code on the installed transformers (get_extended_attention_mask). It is not listed. The recipe's prefix table still names nomic.

🤖 Generated with Claude Code

davanstrien and others added 3 commits November 10, 2026 16:06
The header sets l4x1, a 1h timeout and the HF_TOKEN secret, so the
command needs no job flags. The docstring examples drop them.

Co-Authored-By: Claude Fable 5.2 <noreply@anthropic.com>
google/embeddinggemma-2 failed to load without it: its processor needs
torchvision. With it, the model embeds 100 IMDb rows and uses its
registered document prompt.

Co-Authored-By: Claude Fable 5.2 <noreply@anthropic.com>
A model reached the page only as a script's default value, so
generate-embeddings.py declared one model although it takes any
sentence-transformers model through --model.

TESTED_MODELS lists the models that ran a recipe in a smoke Job with
checked output. Each gets its own entry, with --model set in the command.

Nine embedding models are added after a 100-row smoke Job each. Declared
models go from 30 to 39. nomic-embed-text-v1.5 failed in its own remote
code and is not listed. Records are in maintenance/verifications.jsonl.

Co-Authored-By: Claude Fable 5.2 <noreply@anthropic.com>
@davanstrien
davanstrien merged commit 4652060 into main Nov 10, 2026
1 check passed
@davanstrien
davanstrien deleted the embeddings/declare-models branch November 10, 2026 17:55
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant