* optimize sdv1_5

Author: lxowalle

maix/sdv1_5.py

@@ -500,6 +500,8 @@ def __init__(self, model_dir, backend="axe"):
             raise ValueError("Only support axmodel, please set backend='axe'")
         self.model_suffix = ".axmodel" if backend == "axe" else ".onnx"
         self.model_dir = model_dir
+        if not os.path.exists(model_dir):
+            raise ValueError(f"Model dir {model_dir} not exists")
         self.backend = backend
         self.is_inited = False
 

projects/app_image_generation/main.py

@@ -60,6 +60,10 @@ def __init__(self, disp:display.Display):
             app.set_sys_config_kv("npu", "ai_isp", "0")
 
         # load sdv1_5
+        if not os.path.exists("/root/models/lcm-lora-sdv1-5-maixcam2/ax620e_models"):
+            self.show_error("You need install lcm-lora-sdv1-5-maixcam2 model from https://huggingface.co/sipeed/lcm-lora-sdv1-5-maixcam2")
+            exit(0)
+
         self.__show_load_info('Loading sdv1_5 module..')
         sdv1_5_loader = ModuleLoader('maix.sdv1_5')
         sdv1_5 = None