pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added cuda_is_available (#4725)

Browse Source 
### What problem does this PR solve?

Added cuda_is_available

### Type of change

- [x] Refactoring
This commit is contained in:
Zhichang Yu
2025-02-05 18:01:23 +08:00
committed by GitHub
parent 283d036cba
commit 3411d0a2ce
4 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
uv.lock generated
View File

@ -4814,7 +4814,6 @@ dependencies = [
{ name = "tencentcloud-sdk-python" },
{ name = "tika" },
{ name = "tiktoken" },
{ name = "torch" },
{ name = "umap-learn" },
{ name = "valkey" },
{ name = "vertexai" },
@ -4837,6 +4836,7 @@ full = [
{ name = "fastembed", marker = "platform_machine != 'x86_64' or sys_platform == 'darwin'" },
{ name = "fastembed-gpu", marker = "platform_machine == 'x86_64' and sys_platform != 'darwin'" },
{ name = "flagembedding" },
{ name = "torch" },
{ name = "transformers" },
]
@ -4946,7 +4946,7 @@ requires-dist = [
{ name = "tencentcloud-sdk-python", specifier = "==3.0.1215" },
{ name = "tika", specifier = "==2.6.0" },
{ name = "tiktoken", specifier = "==0.7.0" },
{ name = "torch", specifier = ">=2.5.0,<3.0.0" },
{ name = "torch", marker = "extra == 'full'", specifier = ">=2.5.0,<3.0.0" },
{ name = "transformers", marker = "extra == 'full'", specifier = ">=4.35.0,<5.0.0" },
{ name = "umap-learn", specifier = "==0.5.6" },
{ name = "valkey", specifier = "==6.0.2" },