From 2d17e8df8d65f4a0212ef4c736ec5310dda85d07 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sun, 11 Aug 2024 22:56:51 -0700
Subject: [PATCH] better print

---
 backend/loader.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/backend/loader.py b/backend/loader.py
index b0a8da40..fd06688e 100644
--- a/backend/loader.py
+++ b/backend/loader.py
@@ -177,7 +177,11 @@ def split_state_dict(sd, sd_vae=None):
 
 @torch.no_grad()
 def forge_loader(sd, sd_vae=None):
-    state_dicts, estimated_config = split_state_dict(sd, sd_vae=sd_vae)
+    try:
+        state_dicts, estimated_config = split_state_dict(sd, sd_vae=sd_vae)
+    except:
+        raise ValueError('Failed to recognize model type!')
+    
     repo_name = estimated_config.huggingface_repo
 
     local_path = os.path.join(dir_path, 'huggingface', repo_name)