v 2.0

app.py

@@ -53,7 +53,8 @@ def demo_predict(input_blk, text, num_samples, steps, scale, seed, show_detail):
     cfgs.detailed = show_detail
     seed_everything(seed)
 
-    sampler = init_sampling(cfgs)
+    sampler.num_steps = steps
+    sampler.guider.scale_value = scale
     
     image = input_blk["image"]
     mask = input_blk["mask"]
@@ -128,6 +129,7 @@ if __name__ == "__main__":
     cfgs = OmegaConf.load("./configs/demo.yaml")
 
     model = init_model(cfgs)
+    sampler = init_sampling(cfgs)
     global_index = 0
 
     block = gr.Blocks().queue()

sgm/modules/diffusionmodules/guiders.py

@@ -11,8 +11,8 @@ class VanillaCFG:
     """
 
     def __init__(self, scale, dyn_thresh_config=None):
-        scale_schedule = lambda scale, sigma: scale  # independent of step
-        self.scale_schedule = partial(scale_schedule, scale)
+
+        self.scale_value = scale
         self.dyn_thresh = instantiate_from_config(
             default(
                 dyn_thresh_config,
@@ -24,8 +24,7 @@ class VanillaCFG:
 
     def __call__(self, x, sigma):
         x_u, x_c = x.chunk(2)
-        scale_value = self.scale_schedule(sigma)
-        x_pred = self.dyn_thresh(x_u, x_c, scale_value)
+        x_pred = self.dyn_thresh(x_u, x_c, self.scale_value)
         return x_pred
 
     def prepare_inputs(self, x, s, c, uc):