Upload multit2i.py

multit2i.py

@@ -110,8 +110,7 @@ def get_t2i_model_info_dict(repo_id: str):
 
 
 def rename_image(image_path: str | None, model_name: str, save_path: str | None = None):
-    from PIL import Image, ImageFile
-    ImageFile.LOAD_TRUNCATED_IMAGES = True
+    from PIL import Image
     from datetime import datetime, timezone, timedelta
     if image_path is None: return None
     dt_now = datetime.now(timezone(timedelta(hours=9)))
@@ -174,7 +173,7 @@ def load_from_model(model_name: str, hf_token: str | Literal[False] | None = Non
             if "429" in str(e):
                 raise TooManyRequestsError() from e
         except Exception as e:
-            raise Exception(e)
+            raise Exception() from e
         return data
 
     interface_info = {
@@ -385,7 +384,7 @@ def infer_body(client: InferenceClient | gr.Interface | object, prompt: str, neg
         return str(Path(png_path).resolve())
     except Exception as e:
         print(e)
-        raise Exception(e)
+        raise Exception() from e
 
 
 async def infer(model_name: str, prompt: str, neg_prompt: str | None = None,
@@ -410,12 +409,12 @@ async def infer(model_name: str, prompt: str, neg_prompt: str | None = None,
         print(f"Task timed out: {model_name}")
         if not task.done(): task.cancel()
         result = None
-        raise Exception(f"Task timed out: {model_name}")
+        raise Exception(f"Task timed out: {model_name}") from e
     except Exception as e:
         print(e)
         if not task.done(): task.cancel()
         result = None
-        raise Exception(e)
+        raise Exception() from e
     if task.done() and result is not None:
         with lock:
             image = rename_image(result, model_name, save_path)