Move beta_schedule to model_config and allow disabling unet creation.

This commit is contained in:
comfyanonymous
2023-08-29 14:22:53 -04:00
parent 968078b149
commit 15adc3699f
2 changed files with 4 additions and 2 deletions

View File

@@ -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):