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

Refa: rm useless code. (#11238)

Browse Source 
### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu
2025-11-13 09:59:55 +08:00
committed by GitHub
parent bd5dda6b10
commit ba71160b14
4 changed files with 7 additions and 785 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
deepdoc/vision/ocr.py
View File

@ -117,7 +117,6 @@ def load_model(model_dir, nm, device_id: int | None = None):
providers=['CUDAExecutionProvider'],
provider_options=[cuda_provider_options]
)
run_options.add_run_config_entry("memory.enable_memory_arena_shrinkage", "gpu:" + str(provider_device_id))
logging.info(f"load_model {model_file_path} uses GPU (device {provider_device_id}, gpu_mem_limit={cuda_provider_options['gpu_mem_limit']}, arena_strategy={arena_strategy})")
else:
sess = ort.InferenceSession(