diff --git a/lora_info.py b/lora_info.py index 3c2a332..8fad18b 100644 --- a/lora_info.py +++ b/lora_info.py @@ -61,8 +61,11 @@ def get_lora_info(lora_name): lora_path = folder_paths.get_full_path("loras", lora_name) LORAsha256 = calculate_sha256(lora_path) model_info = get_model_version_info(LORAsha256) - trainedWords = ",".join(model_info.get("trainedWords")) - baseModel = model_info.get("baseModel") + if model_info.get("trainedWords", None) is None: + trainedWords = "" + else: + trainedWords = ",".join(model_info.get("trainedWords")) + baseModel = model_info.get("baseModel", "") images = model_info.get('images') examplePrompt = None