Spaces:
Sleeping
Sleeping
Update app
Browse files- scripts/build_index.py +2 -0
scripts/build_index.py
CHANGED
|
@@ -14,6 +14,7 @@ from safetensors.numpy import save_file as safetensors_save
|
|
| 14 |
|
| 15 |
|
| 16 |
ROOT = Path(__file__).resolve().parent.parent
|
|
|
|
| 17 |
|
| 18 |
|
| 19 |
def load_detector(transformers_dir: Path):
|
|
@@ -111,6 +112,7 @@ def main() -> None:
|
|
| 111 |
raise SystemExit("Expected a transformers clone at ./transformers or ./transformers_repo")
|
| 112 |
|
| 113 |
detector = load_detector(transformers_dir)
|
|
|
|
| 114 |
hub_dataset = os.getenv("HUB_DATASET", detector.HUB_DATASET_DEFAULT)
|
| 115 |
analyzer = detector.CodeSimilarityAnalyzer(hub_dataset=hub_dataset)
|
| 116 |
analyzer.models_root = (transformers_dir / "src" / "transformers" / "models").resolve()
|
|
|
|
| 14 |
|
| 15 |
|
| 16 |
ROOT = Path(__file__).resolve().parent.parent
|
| 17 |
+
EMBEDDING_MODEL = "Qwen/Qwen3-Embedding-0.6B"
|
| 18 |
|
| 19 |
|
| 20 |
def load_detector(transformers_dir: Path):
|
|
|
|
| 112 |
raise SystemExit("Expected a transformers clone at ./transformers or ./transformers_repo")
|
| 113 |
|
| 114 |
detector = load_detector(transformers_dir)
|
| 115 |
+
detector.EMBEDDING_MODEL = EMBEDDING_MODEL
|
| 116 |
hub_dataset = os.getenv("HUB_DATASET", detector.HUB_DATASET_DEFAULT)
|
| 117 |
analyzer = detector.CodeSimilarityAnalyzer(hub_dataset=hub_dataset)
|
| 118 |
analyzer.models_root = (transformers_dir / "src" / "transformers" / "models").resolve()
|