diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index c0b68fb8c..deb4aa4c2 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -275,6 +275,9 @@ class ModelPatcher: self.size = comfy.model_management.module_size(self.model) return self.size + def get_ram_usage(self): + return self.model_size() + def loaded_size(self): return self.model.model_loaded_weight_memory