Better per model memory usage estimations.

2024-08-02 18:08:21 -04:00
parent 3a9ee995cf
commit ea03c9dcd2
3 changed files with 17 additions and 25 deletions
--- a/comfy/supported_models_base.py
+++ b/comfy/supported_models_base.py
@@ -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