mirror of
https://github.com/asg017/sqlite-vec.git
synced 2026-04-25 08:46:49 +02:00
8544081a67
Extend benchmarks-ann/ with results database (SQLite with per-query detail and continuous writes), dataset subfolder organization, --subset-size and --warmup options. Supports systematic comparison across flat, rescore, IVF, and DiskANN index types.
13 lines
327 B
Python
13 lines
327 B
Python
# /// script
|
|
# requires-python = ">=3.12"
|
|
# dependencies = [
|
|
# "model2vec[distill]",
|
|
# "torch<=2.7",
|
|
# ]
|
|
# ///
|
|
|
|
from model2vec.distill import distill
|
|
|
|
model = distill(model_name="BAAI/bge-base-en-v1.5", pca_dims=768)
|
|
model.save_pretrained("bge-base-en-v1.5-768")
|
|
print("Saved distilled model to bge-base-en-v1.5-768/")
|