v0.2.1

install.py

@@ -1,11 +1,14 @@
 import launch
 import sys
-from importlib_metadata import version
 
 python = sys.executable
 
 
 def install():
+    if not launch.is_installed("importlib_metadata"):
+        launch.run_pip("install importlib_metadata", "importlib_metadata", live=True)
+    from importlib_metadata import version
+
     if launch.is_installed("tensorrt"):
         if not version("tensorrt") == "9.0.1.post11.dev4":
             launch.run(

scripts/trt.py

@@ -299,7 +299,7 @@ def process_batch(self, p, *args, **kwargs):
         if self.torch_unet:
             return super().process_batch(p, *args, **kwargs)
 
-        if self.idx != sd_unet.current_unet.profile_idx:
+        if sd_unet.current_unet is not None and self.idx != sd_unet.current_unet.profile_idx:
             sd_unet.current_unet.profile_idx = self.idx
             sd_unet.current_unet.switch_engine()
 

ui_trt.py

@@ -268,7 +268,10 @@ def get_lora_checkpoints():
         if os.path.exists(config_file):
             with open(config_file, "r") as f:
                 config = json.load(f)
-            version = SDVersion.from_str(config["sd version"])
+            try:
+                version = SDVersion.from_str(config["sd version"])
+            except:
+                version = SDVersion.Unknown
 
         else:
             version = SDVersion.Unknown