Skip to content

Commit 9a23498

Browse files
committed
catch get_lora_checkpoints() exceptions
1 parent 9e9b21f commit 9a23498

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

ui_trt.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -557,10 +557,14 @@ def get_lora_checkpoints():
557557
)
558558
for filename in canditates:
559559
name = os.path.splitext(os.path.basename(filename))[0]
560-
metadata = sd_models.read_metadata_from_safetensors(filename)
560+
try:
561+
metadata = sd_models.read_metadata_from_safetensors(filename)
562+
version = get_version_from_filename(metadata.get("ss_sd_model_name"))
563+
except (AssertionError, TypeError):
564+
version = "Unknown"
561565
available_lora_models[name] = {
562566
"filename": filename,
563-
"version": get_version_from_filename(metadata.get("ss_sd_model_name")),
567+
"version": version,
564568
}
565569
return available_lora_models
566570

0 commit comments

Comments
 (0)