LPS-36216 Use the proper StagedModelTypes for DL