fix cache globals
This commit is contained in:
parent
ee495a8d93
commit
c7fa50a028
|
@ -22,6 +22,9 @@ last_pipeline_params = None
|
|||
|
||||
|
||||
def load_gfpgan(ctx: ServerContext, upscale: UpscaleParams):
|
||||
global last_pipeline_instance
|
||||
global last_pipeline_params
|
||||
|
||||
if upsampler is None:
|
||||
upsampler = load_resrgan(ctx, upscale)
|
||||
|
||||
|
|
|
@ -23,6 +23,9 @@ last_pipeline_params = (None, None)
|
|||
|
||||
|
||||
def load_resrgan(ctx: ServerContext, params: UpscaleParams, tile=0):
|
||||
global last_pipeline_instance
|
||||
global last_pipeline_params
|
||||
|
||||
model_file = '%s.%s' % (params.upscale_model, params.format)
|
||||
model_path = path.join(ctx.model_path, model_file)
|
||||
if not path.isfile(model_path):
|
||||
|
|
|
@ -26,6 +26,9 @@ last_pipeline_params = (None, None)
|
|||
|
||||
|
||||
def load_stable_diffusion(ctx: ServerContext, upscale: UpscaleParams):
|
||||
global last_pipeline_instance
|
||||
global last_pipeline_params
|
||||
|
||||
model_path = path.join(ctx.model_path, upscale.upscale_model)
|
||||
cache_params = (model_path, upscale.format)
|
||||
|
||||
|
|
Loading…
Reference in New Issue