Molbap HF Staff commited on
Commit
afd3562
·
verified ·
1 Parent(s): 430f700

Update app

Browse files
Files changed (1) hide show
  1. 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()