refine

siliconflow · Aug 5, 2024 · 2c006eb · 2c006eb
1 parent 0083d96
commit 2c006eb
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 6 deletions.
diff --git a/nodes.py b/nodes.py
@@ -464,7 +464,7 @@ def INPUT_TYPES(s):
         return {
             "required": {
                 "unet_name": (folder_paths.get_filename_list("unet"),),
-                "weight_dtype": (["fp8_e4m3fn"],),
+                "weight_dtype": (["default"],),
             }
         }
 

diff --git a/src/bizyair/configs/models.json b/src/bizyair/configs/models.json
@@ -47,7 +47,8 @@
       "Kolors.safetensors"
     ],
     "flux": [
-      "flux1-schnell.sft"
+      "flux1-schnell.sft",
+      "flux1-dev.sft"
     ]
   },
   "vae": {
@@ -60,6 +61,6 @@
   },
   "clip":[
     "clip_l.safetensors",
-    "t5xxl_fp8_e4m3fn.safetensors"
+    "t5xxl_fp16.safetensors"
   ]
 }
diff --git a/src/bizyair/image_utils.py b/src/bizyair/image_utils.py
@@ -253,9 +253,13 @@ def send_request(
                 raise RuntimeError(
                     f'Unexpected error accessing result["data"]["payload"]. Result: {result}'
                 ) from e
-
-        real_out = decode_data(out)
-        return real_out[0]
+        try:
+            real_out = decode_data(out)
+            return real_out[0]
+        except Exception as e:
+            raise RuntimeError(
+                f"Exception: {e=} {self._short_repr(out, max_length=100)}"
+            ) from e
 
 
 def convert_image_to_rgb(image: Image.Image) -> Image.Image: