From 95a78f629588b6466375619506eadd7b87fe3c58 Mon Sep 17 00:00:00 2001
From: Michael Bakker <michael.bakker1986@gmail.com>
Date: Tue, 6 Feb 2024 21:38:17 +0100
Subject: [PATCH] Update README.md

bugfix in set_scheduler(a, b) takes one argument but requires 2
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 202a6c9..26f1768 100644
--- a/README.md
+++ b/README.md
@@ -68,7 +68,7 @@ stage_3.enable_xformers_memory_efficient_attention()  # remove line if torch.__v
 stage_3.enable_model_cpu_offload()
 
 
-def set_scheduler(stage):
+def set_scheduler(stage, scheduler_name):
     if scheduler_name == 'dpm++':
         scheduler = DPMSolverMultistepScheduler.from_config(stage.scheduler.config)
         scheduler.config.algorithm_type = 'dpmsolver++'