nodes/city96-ComfyUI-GGUF
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

HiDream support

Browse Source 
#248
This commit is contained in:
City
2025-04-16 01:43:15 +02:00
parent 794fc28536
commit 2ef5a66a9f
3 changed files with 46 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
loader.py
+1 -1
View File
@@ -7,7 +7,7 @@ import gguf
from .ops import GGMLTensor
from .dequant import is_quantized, dequantize_tensor
IMG_ARCH_LIST = {"flux", "sd1", "sdxl", "sd3", "aura", "ltxv", "hyvid", "wan"}
IMG_ARCH_LIST = {"flux", "sd1", "sdxl", "sd3", "aura", "hidream", "ltxv", "hyvid", "wan"}
TXT_ARCH_LIST = {"t5", "t5encoder", "llama"}
def get_orig_shape(reader, tensor_name):