search

Unstructured-IO / unstructured

Open source libraries and APIs to build custom preprocessing pipelines for labeling, training, or production machine learning pipelines.
https://www.unstructured.io/
Apache License 2.0
7.4k stars 573 forks source link

GPU #3241

Closed Deh-alba closed 49 minutes ago

Deh-alba commented 1 week ago

Hey guys, we have the possibility to run the unstructured in GPU, could be nice and will help a lot

MthwRobinson commented 49 minutes ago

Hi @Deh-alba - as of now, we don't plan to implement strong support for GPUs in unstructured. However, the YOLOX ONNX model in unstructured-inference is configured to look for the CUDA provider when available.

https://github.com/Unstructured-IO/unstructured-inference/blob/97cbcb35f54475a75647c22f92a6786f1828dbbe/unstructured_inference/models/yolox.py#L76-L81