Allow custom ServingRuntimes

Signed-off-by: Salomon Popp <[email protected]>
kserve · Jun 26, 2024 · 3a8c824 · 3a8c824
1 parent 3999015
commit 3a8c824
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts
@@ -143,14 +143,7 @@ export function getPredictorExtensionSpec(
   predictor: PredictorSpec,
 ): PredictorExtensionSpec {
   if (predictor.model) {
-    if (
-      Object.values(PredictorType).includes(
-        predictor.model?.modelFormat.name as PredictorType,
-      )
-    ) {
-      const spec = predictor.model;
-      return spec;
-    }
+    return predictor.model
   } else {
     for (const predictorType of Object.values(PredictorType)) {
       if (predictorType in predictor) {