From d94998d73bd0f50e5c97a786b5fc48c0fc87bfdf Mon Sep 17 00:00:00 2001 From: cocktailpeanut Date: Fri, 17 Jan 2025 21:11:42 -0500 Subject: [PATCH] set use_kv_cache to false if no offload_model --- app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app.py b/app.py index ba87673..a0d26f7 100644 --- a/app.py +++ b/app.py @@ -33,7 +33,7 @@ def generate_image(text, img1, img2, img3, height, width, guidance_scale, img_gu img_guidance_scale=img_guidance_scale, num_inference_steps=inference_steps, separate_cfg_infer=separate_cfg_infer, - use_kv_cache=True, + use_kv_cache=(True if offload_model else False), offload_kv_cache=True, offload_model=offload_model, use_input_image_size_as_output=use_input_image_size_as_output,