Can be done after #477 This is a follow-up for https://github.com/foundation-model-stack/foundation-model-stack/issues/461
Can be done after #477
This is a follow-up for #461