unsloth/studio/backend
Roland Tannous 5a086353ab feat: add embedding model training support
Add end-to-end embedding/sentence-transformer training pipeline using
FastSentenceTransformer, SentenceTransformerTrainer, and
MultipleNegativesRankingLoss with BatchSamplers.NO_DUPLICATES.

Backend:
- Add is_embedding_model() detection via HF tags + pipeline_tag
- Add /check-embedding/ API route and EmbeddingCheckResponse
- Extend derive_model_type() to return "embeddings"
- Add _run_embedding_training() in worker.py with progress callbacks,
  stop handling, LoRA (task_type=FEATURE_EXTRACTION), and model saving
- Add is_embedding field to TrainingStartRequest and ModelDetails
- Add YAML configs for 5 models: all-MiniLM-L6-v2, bge-m3,
  embeddinggemma-300m, gte-modernbert-base, Qwen3-Embedding-0.6B

Frontend:
- Wire isEmbeddingModel flag through store, API types, and mappers
- Force packing=false, train_on_completions=false, warmup_ratio=0.03
- Hide packing and train_on_completions checkboxes for embedding models
- Auto-set modelType to "embeddings" from backend model_type response
2026-03-10 18:10:09 +00:00
..
assets feat: add embedding model training support 2026-03-10 18:10:09 +00:00
auth Revert "Merge pull request #347 from unslothai/feature/studio-storage-roots" 2026-03-10 01:52:47 +00:00
core feat: add embedding model training support 2026-03-10 18:10:09 +00:00
loggers root studio folder 2026-02-02 09:13:49 +00:00
models feat: add embedding model training support 2026-03-10 18:10:09 +00:00
plugins/data-designer-unstructured-seed Revert "Merge pull request #347 from unslothai/feature/studio-storage-roots" 2026-03-10 01:52:47 +00:00
requirements chore(data-recipe): bump data-designer to 0.5.2 and pin duckdb<1.5 2026-03-09 17:27:02 +01:00
routes feat: add embedding model training support 2026-03-10 18:10:09 +00:00
state root studio folder 2026-02-02 09:13:49 +00:00
tests Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00
utils feat: add embedding model training support 2026-03-10 18:10:09 +00:00
colab.py Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00
main.py Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00
run.py Add AGPL-3.0 SPDX headers to all source files 2026-03-09 20:17:45 +00:00