AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-10 21:49:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix:Set CUDA_VISIBLE_DEVICES In DefaultEmbedding (#7465)

Browse Source 
### What problem does this PR solve?
https://github.com/infiniflow/ragflow/issues/7420

### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Stephen Hu 2025-05-06 14:38:36 +08:00 committed by GitHub
parent 60787f8d5d
commit 65537b8200
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
rag/llm/embedding_model.py
View File

@ -57,6 +57,7 @@ class Base(ABC):
class DefaultEmbedding(Base): class DefaultEmbedding(Base):
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
_model = None _model = None
_model_name = "" _model_name = ""
_model_lock = threading.Lock() _model_lock = threading.Lock()