hailin
/
embed-bge-m3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-05 17:08:08 +08:00
parent 0b75bf040e
commit 76e7594895
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/main.py
View File

@ -182,7 +182,7 @@ logger.info("Using SAFE_MIN_FREE_MB = %d MB", SAFE_MIN_FREE_MB)
def _warm_worker(t, q): def _warm_worker(t, q):
try: try:
_ = model.encode(t, return_dense=True, num_processes=1) _ = model.encode(t, return_dense=True)
q.put("ok") q.put("ok")
except Exception as e: except Exception as e:
q.put(str(e)) q.put(str(e))
@ -216,7 +216,7 @@ class EmbeddingRequest(BaseModel):
def _worker(t, q): def _worker(t, q):
try: try:
# out = model.encode(t, return_dense=True) # GPU or CPU 均安全 # out = model.encode(t, return_dense=True) # GPU or CPU 均安全
out = model.encode(t, return_dense=True, num_processes=1) out = model.encode(t, return_dense=True)
q.put(("ok", out)) q.put(("ok", out))
except Exception as e: except Exception as e:
q.put(("err", str(e))) q.put(("err", str(e)))