fix key error for custom models (#12042)

This commit is contained in:
Marc Altmann
2024-06-18 07:40:54 -06:00
committed by GitHub
parent bdda89b5e2
commit 4bca405e29