rishik1111 commited on
Commit
a4f899d
·
1 Parent(s): 2a8bb45

feat: Implement lazy module loading for zero-overhead server startup

Browse files
vector_search/embed.py CHANGED
@@ -12,17 +12,10 @@ import os
12
  from pathlib import Path
13
  from typing import List, Union
14
  import numpy as np
15
- import torch
16
  import config
17
 
18
  logger = logging.getLogger(__name__)
19
 
20
- # Optimize PyTorch CPU parallelism
21
- try:
22
- torch.set_num_threads(max(1, torch.get_num_threads()))
23
- except Exception:
24
- pass
25
-
26
  _EMBEDDER_INSTANCE = None
27
 
28
 
 
12
  from pathlib import Path
13
  from typing import List, Union
14
  import numpy as np
 
15
  import config
16
 
17
  logger = logging.getLogger(__name__)
18
 
 
 
 
 
 
 
19
  _EMBEDDER_INSTANCE = None
20
 
21
 
vector_search/rerank.py CHANGED
@@ -4,9 +4,7 @@ import re
4
  from pathlib import Path
5
  from typing import Any, Dict, List
6
  import numpy as np
7
- import torch
8
  from rank_bm25 import BM25Okapi
9
- from sentence_transformers.cross_encoder import CrossEncoder
10
  import config
11
 
12
  logger = logging.getLogger(__name__)
@@ -180,6 +178,7 @@ class PyTorchCrossEncoderRanker:
180
  PyTorch fallback wrapper for sentence-transformers CrossEncoder.
181
  """
182
  def __init__(self, model_name: str = config.CROSS_ENCODER_MODEL_NAME):
 
183
  load_path = model_name
184
  local_cache = getattr(config, "CROSS_ENCODER_LOCAL_CACHE", None)
185
  if local_cache and Path(local_cache).exists():
 
4
  from pathlib import Path
5
  from typing import Any, Dict, List
6
  import numpy as np
 
7
  from rank_bm25 import BM25Okapi
 
8
  import config
9
 
10
  logger = logging.getLogger(__name__)
 
178
  PyTorch fallback wrapper for sentence-transformers CrossEncoder.
179
  """
180
  def __init__(self, model_name: str = config.CROSS_ENCODER_MODEL_NAME):
181
+ from sentence_transformers.cross_encoder import CrossEncoder
182
  load_path = model_name
183
  local_cache = getattr(config, "CROSS_ENCODER_LOCAL_CACHE", None)
184
  if local_cache and Path(local_cache).exists():