From c5dcfc1f1379deba22e96d7c47f7bec5e33c954c Mon Sep 17 00:00:00 2001 From: Gaurav Shukla Date: Fri, 8 Sep 2023 23:00:29 +0530 Subject: [PATCH] [vicuna] Exit when mlir is not present in shark tank (#1825) Signed-off-by: Gaurav Shukla --- apps/language_models/scripts/vicuna.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/apps/language_models/scripts/vicuna.py b/apps/language_models/scripts/vicuna.py index 0c5c2c335d..c40c6148bd 100644 --- a/apps/language_models/scripts/vicuna.py +++ b/apps/language_models/scripts/vicuna.py @@ -7,6 +7,7 @@ from tqdm import tqdm from typing import List, Tuple import subprocess +import sys import torch import torch_mlir @@ -1434,6 +1435,8 @@ def compile(self): print(f"[DEBUG] mlir not found") # Disabling this path of IR generation for now as it is broken. print("Please check if the mlir file is present at the shark tank. Exiting.") + self.shark_model = None + sys.exit() return print("[DEBUG] generating mlir on device")