wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-29 02:17:52 +08:00
Code Issues Actions 30 Packages Projects Releases Wiki Activity

Closer memory usage factors for PID (#14123)

Browse Source
This commit is contained in:
comfyanonymous 2026-05-26 18:04:55 -07:00 committed by GitHub
parent 28f4ef277c
commit d8d860a588
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/supported_models.py
View File

@ -1214,7 +1214,7 @@ class PixelDiTT2I(supported_models_base.BASE):
} }
latent_format = latent_formats.PixelDiTPixel latent_format = latent_formats.PixelDiTPixel
memory_usage_factor = 0.18 memory_usage_factor = 0.04
supported_inference_dtypes = [torch.bfloat16, torch.float32] supported_inference_dtypes = [torch.bfloat16, torch.float32]
vae_key_prefix = ["vae."] vae_key_prefix = ["vae."]
@ -1263,7 +1263,7 @@ class PiD(PixelDiTT2I):
"shift": 1.5, # close approximation of the original distill 4 steps [0.999, 0.866, 0.634, 0.342, 0] "shift": 1.5, # close approximation of the original distill 4 steps [0.999, 0.866, 0.634, 0.342, 0]
} }
memory_usage_factor = 0.07 memory_usage_factor = 0.04
def get_model(self, state_dict, prefix="", device=None): def get_model(self, state_dict, prefix="", device=None):
return model_base.PiD(self, device=device) return model_base.PiD(self, device=device)