Procházet zdrojové kódy

Fix:Set CUDA_VISIBLE_DEVICES In DefaultEmbedding (#7465)

### 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)
tags/v0.19.0
Stephen Hu před 6 měsíci
rodič
revize
65537b8200
Žádný účet není propojen s e-mailovou adresou tvůrce revize
1 změnil soubory, kde provedl 1 přidání a 0 odebrání
  1. 1
    0
      rag/llm/embedding_model.py

+ 1
- 0
rag/llm/embedding_model.py Zobrazit soubor

@@ -57,6 +57,7 @@ class Base(ABC):


class DefaultEmbedding(Base):
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
_model = None
_model_name = ""
_model_lock = threading.Lock()

Načítá se…
Zrušit
Uložit