• myhloli's avatar
    refactor(device): optimize memory cleaning and device selection · 50f48417
    myhloli authored
    - Update clean_memory function to support both CUDA and NPU devices
    - Implement get_device function to centralize device selection logic
    - Modify model initialization and memory cleaning to use the selected device
    - Update RapidTableModel to support both RapidOCR and PaddleOCR engines
    50f48417
model_utils.py 2.42 KB