From 32447f0c392be6a6b64fbac09fd7e7f33eb451f8 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Wed, 22 Nov 2023 17:23:37 -0500
Subject: [PATCH] Add sampling_settings so models can specify specific sampling
 settings.

---
 comfy/model_sampling.py        | 2 +-
 comfy/supported_models_base.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/comfy/model_sampling.py b/comfy/model_sampling.py
index 37a3ac72..9e2a1c1a 100644
--- a/comfy/model_sampling.py
+++ b/comfy/model_sampling.py
@@ -24,7 +24,7 @@ class ModelSamplingDiscrete(torch.nn.Module):
         super().__init__()
         beta_schedule = "linear"
         if model_config is not None:
-            beta_schedule = model_config.beta_schedule
+            beta_schedule = model_config.sampling_settings.get("beta_schedule", beta_schedule)
         self._register_schedule(given_betas=None, beta_schedule=beta_schedule, timesteps=1000, linear_start=0.00085, linear_end=0.012, cosine_s=8e-3)
         self.sigma_data = 1.0
 
diff --git a/comfy/supported_models_base.py b/comfy/supported_models_base.py
index b073eb4f..3412cfea 100644
--- a/comfy/supported_models_base.py
+++ b/comfy/supported_models_base.py
@@ -19,7 +19,7 @@ class BASE:
     clip_prefix = []
     clip_vision_prefix = None
     noise_aug_config = None
-    beta_schedule = "linear"
+    sampling_settings = {}
     latent_format = latent_formats.LatentFormat
 
     @classmethod