apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix cache globals

This commit is contained in:
Sean Sube 2023-01-28 00:08:52 -06:00
parent ee495a8d93
commit c7fa50a028
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/chain/correct_gfpgan.py
View File

@ -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)

3
api/onnx_web/chain/upscale_resrgan.py
View File

@ -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):

3
api/onnx_web/chain/upscale_stable_diffusion.py
View File

@ -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)