Molbap HF Staff commited on
Commit
06274f5
·
1 Parent(s): b90f4fc
Files changed (1) hide show
  1. modular_graph_and_candidates.py +2 -2
modular_graph_and_candidates.py CHANGED
@@ -97,7 +97,7 @@ def similarity_clusters(bags: Dict[str, List[Set[str]]], thr: float) -> Dict[Tup
97
 
98
  @spaces.GPU
99
  def embedding_similarity_clusters(models_root: Path, missing: List[str], thr: float) -> Dict[Tuple[str, str], float]:
100
- model = SentenceTransformer("codesage/codesage-large-v2", device="cpu", trust_remote_code=True)
101
 
102
  try:
103
  cfg = model[0].auto_model.config
@@ -339,7 +339,7 @@ function updateVisibility() {
339
  }
340
  document.getElementById('toggleRed').addEventListener('change', updateVisibility);
341
 
342
- const HF_LOGO_URI = "__HF_LOGO_DATA_URI__";
343
  const graph = __GRAPH_DATA__;
344
  const W = innerWidth, H = innerHeight;
345
  const svg = d3.select('#dependency').call(d3.zoom().on('zoom', e => g.attr('transform', e.transform)));
 
97
 
98
  @spaces.GPU
99
  def embedding_similarity_clusters(models_root: Path, missing: List[str], thr: float) -> Dict[Tuple[str, str], float]:
100
+ model = SentenceTransformer("codesage/codesage-large-v2", device="cuda", trust_remote_code=True)
101
 
102
  try:
103
  cfg = model[0].auto_model.config
 
339
  }
340
  document.getElementById('toggleRed').addEventListener('change', updateVisibility);
341
 
342
+ const HF_LOGO_URI = "hf-logo.png";
343
  const graph = __GRAPH_DATA__;
344
  const W = innerWidth, H = innerHeight;
345
  const svg = d3.select('#dependency').call(d3.zoom().on('zoom', e => g.attr('transform', e.transform)));