refactor: Remove unnecessary device selection in predict function be3fb72 yangliz5 commited on Oct 11, 2024