Fix for https://github.com/comfyanonymous/ComfyUI/issues/2038

2023-12-02 09:43:44 +07:00
parent 77dd7dbb91
commit 4dc7fa8f39
1 changed files with 2 additions and 1 deletions
@@ -20,7 +20,8 @@ SLIDING_CONTEXT_LENGTH = 16

 class ModelSamplingConfig:
    def __init__(self, beta_schedule: str):
-        self.beta_schedule = beta_schedule
+        self.sampling_settings = {}
+        self.sampling_settings['beta_schedule'] = beta_schedule


 def forward_timestep_embed(ts, x, emb, context=None, transformer_options={}, output_shape=None):