mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-10 19:46:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Better per model memory usage estimations.

Browse Source
This commit is contained in:
comfyanonymous
2024-08-02 18:08:21 -04:00
parent 3a9ee995cf
commit ea03c9dcd2
3 changed files with 17 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/supported_models_base.py
View File

@@ -27,6 +27,8 @@ class BASE:
text_encoder_key_prefix = ["cond_stage_model."]
supported_inference_dtypes = [torch.float16, torch.bfloat16, torch.float32]
memory_usage_factor = 2.0
manual_cast_dtype = None
@classmethod