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,