Hey @philschmid thanks for your response. I’ve experienced this issue with the hosted inference API but suspect it might be the same with Amazon SageMaker as well they both use the pipeline model if I’m not mistaken?