From dd713c24cd723bb77277f3b86f2c4cf447b4c62e Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Sat, 16 Dec 2023 01:23:34 +0100
Subject: [PATCH] fixed errors

---
 lollms/config.py                      | 2 ++
 lollms/image_gen_modules/lollms_sd.py | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/lollms/config.py b/lollms/config.py
index ef0edcb..f76c2d7 100644
--- a/lollms/config.py
+++ b/lollms/config.py
@@ -542,6 +542,8 @@ class TypedConfig:
                     entry_value = bool(entry_value)
                 elif entry_type == "list":
                     entry_value = list(entry_value)
+                elif entry_type == "dict":
+                    entry_value = eval(entry_value)
                 else:
                     raise ValueError(f"Invalid field type '{entry_type}' for entry '{entry_name}'.")
 
diff --git a/lollms/image_gen_modules/lollms_sd.py b/lollms/image_gen_modules/lollms_sd.py
index 68f3bc3..7faa4e1 100644
--- a/lollms/image_gen_modules/lollms_sd.py
+++ b/lollms/image_gen_modules/lollms_sd.py
@@ -344,7 +344,7 @@ class LollmsSD:
                 trace_exception(ex)  
 
 
-        return img_paths[0], infos
+        return img_paths[0] if len(img_paths)>0 else None, infos
 
     def check_controlnet(self):
         try: