fix: gate onnx and tensorrt-llm run on darwin

janhq · Jul 4, 2024 · 7073988 · 7073988
1 parent e2b5d5d
commit 7073988
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/cortex-js/src/infrastructure/commanders/shortcuts/run.command.ts b/cortex-js/src/infrastructure/commanders/shortcuts/run.command.ts
@@ -54,6 +54,10 @@ export class RunCommand extends CommandRunner {
         exit(1);
       }
     }
+
+    // Check model compatibility on this machine
+    await checkModelCompatibility(modelId);
+
     // If not exist
     // Try Pull
     if (!(await this.modelsCliUsecases.getModel(modelId))) {
@@ -73,16 +77,11 @@ export class RunCommand extends CommandRunner {
       !Array.isArray(existingModel.files) ||
       /^(http|https):\/\/[^/]+\/.*/.test(existingModel.files[0])
     ) {
-      checkingSpinner.fail(
-        `Model is not available`
-      );
+      checkingSpinner.fail(`Model is not available`);
       process.exit(1);
     }
     checkingSpinner.succeed('Model found');
 
-    // Check model compatibility on this machine
-    await checkModelCompatibility(modelId);
-
     const engine = existingModel.engine || Engines.llamaCPP;
     // Pull engine if not exist
     if (