新的 API 始终使用 GPU

作者: darshats创建于 2023年4月25日更新于 2026年7月9日
标签bug

在 inference.py 第 193 行,模型类 (新的 API) 调用了较旧的 predict 方法,但跳过了传入设备。因此,默认情况下总是使用 cuda 设备,尽管 model.device 可能是 cpu。

内容来源: IDEA-Research/GroundingDINO