mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-11 03:58:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Basic support for hidream i1 model.

Browse Source
This commit is contained in:
comfyanonymous
2025-04-15 17:35:05 -04:00
parent 6fc5dbd52a
commit 9ad792f927
9 changed files with 1087 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ops.py
View File

@@ -263,6 +263,9 @@ class manual_cast(disable_weight_init):
class ConvTranspose1d(disable_weight_init.ConvTranspose1d):
comfy_cast_weights = True
class RMSNorm(disable_weight_init.RMSNorm):
comfy_cast_weights = True
class Embedding(disable_weight_init.Embedding):
comfy_cast_weights = True