jaidaken/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix chroma fp8 te being treated as fp16. (#11795)

Browse Source
This commit is contained in:
comfyanonymous
2026-01-10 14:40:42 -08:00
committed by GitHub
parent cd912963f1
commit 2f642d5d9b
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/text_encoders/cosmos.py
View File

@@ -36,7 +36,7 @@ def te(dtype_t5=None, t5_quantization_metadata=None):
if t5_quantization_metadata is not None: if t5_quantization_metadata is not None:
model_options = model_options.copy() model_options = model_options.copy()
model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata
if dtype is None: if dtype_t5 is not None:
dtype = dtype_t5 dtype = dtype_t5
super().__init__(device=device, dtype=dtype, model_options=model_options) super().__init__(device=device, dtype=dtype, model_options=model_options)
return CosmosTEModel_ return CosmosTEModel_

2
comfy/text_encoders/genmo.py
View File

@@ -32,7 +32,7 @@ def mochi_te(dtype_t5=None, t5_quantization_metadata=None):
if t5_quantization_metadata is not None: if t5_quantization_metadata is not None:
model_options = model_options.copy() model_options = model_options.copy()
model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata
if dtype is None: if dtype_t5 is not None:
dtype = dtype_t5 dtype = dtype_t5
super().__init__(device=device, dtype=dtype, model_options=model_options) super().__init__(device=device, dtype=dtype, model_options=model_options)
return MochiTEModel_ return MochiTEModel_

2
comfy/text_encoders/pixart_t5.py
View File

@@ -36,7 +36,7 @@ def pixart_te(dtype_t5=None, t5_quantization_metadata=None):
if t5_quantization_metadata is not None: if t5_quantization_metadata is not None:
model_options = model_options.copy() model_options = model_options.copy()
model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata model_options["t5xxl_quantization_metadata"] = t5_quantization_metadata
if dtype is None: if dtype_t5 is not None:
dtype = dtype_t5 dtype = dtype_t5
super().__init__(device=device, dtype=dtype, model_options=model_options) super().__init__(device=device, dtype=dtype, model_options=model_options)
return PixArtTEModel_ return PixArtTEModel_