fix key error for custom models (#12042)

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