Move beta_schedule to model_config and allow disabling unet creation.

2023-08-29 14:22:53 -04:00
parent 968078b149
commit 15adc3699f
2 changed files with 4 additions and 2 deletions
--- a/comfy/supported_models_base.py
+++ b/comfy/supported_models_base.py
@@ -33,6 +33,7 @@ class BASE:
    clip_prefix = []
    clip_vision_prefix = None
    noise_aug_config = None
+    beta_schedule = "linear"

    @classmethod
    def matches(s, unet_config):