Allow custom ServingRuntimes

kserve · Nov 7, 2022 · 92a85d3 · 92a85d3
1 parent a3a8e80
commit 92a85d3
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/frontend/src/app/shared/utils.ts b/frontend/src/app/shared/utils.ts
@@ -126,10 +126,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) {