mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-14 11:07:24 +08:00
Better Hidream O1 mem usage factor for non dynamic vram. (#13864)
This commit is contained in:
parent
cccb697aa3
commit
2bd65f2091
@ -1443,7 +1443,7 @@ class HiDreamO1(supported_models_base.BASE):
|
|||||||
}
|
}
|
||||||
|
|
||||||
latent_format = latent_formats.HiDreamO1Pixel
|
latent_format = latent_formats.HiDreamO1Pixel
|
||||||
memory_usage_factor = 0.6
|
memory_usage_factor = 0.033
|
||||||
# fp16 not supported: LM MLP down_proj activations fp16 overflow, causing NaNs
|
# fp16 not supported: LM MLP down_proj activations fp16 overflow, causing NaNs
|
||||||
supported_inference_dtypes = [torch.bfloat16, torch.float32]
|
supported_inference_dtypes = [torch.bfloat16, torch.float32]
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user