Fix img2img mode switch (#1247)

* add updated scheduler value in global config

* clear scheduler global variable with others
This commit is contained in:
Phaneesh Barwaria
2023-03-27 19:31:22 +05:30
committed by GitHub
parent fb35e13e7a
commit 431132b8ee
2 changed files with 5 additions and 2 deletions

View File

@@ -173,7 +173,7 @@ def img2img_inf(
else "stabilityai/stable-diffusion-2-1-base"
)
global_obj.set_schedulers(get_schedulers(model_id))
scheduler_obj = global_obj.get_scheduler(scheduler)
scheduler_obj = global_obj.get_scheduler(args.scheduler)
if use_stencil is not None:
args.use_tuned = False
@@ -218,7 +218,7 @@ def img2img_inf(
)
)
global_obj.set_sd_scheduler(scheduler)
global_obj.set_sd_scheduler(args.scheduler)
start_time = time.time()
global_obj.get_sd_obj().log = ""

View File

@@ -61,8 +61,11 @@ def get_scheduler(key):
def clear_cache():
global _sd_obj
global _config_obj
global _schedulers
del _sd_obj
del _config_obj
del _schedulers
gc.collect()
_sd_obj = None
_config_obj = None
_schedulers = None