diff --git a/demo_gradio.py b/demo_gradio.py
index d98df6dd..d9a3c213 100644
--- a/demo_gradio.py
+++ b/demo_gradio.py
@@ -162,7 +162,7 @@
 image_encoder.requires_grad_(False)
 transformer.requires_grad_(False)
 
-if config["lora"]:
+if config.get("lora", None):
     transformer = load_lora(transformer,  config["lora"]["path"], config["lora"]["name"])
 
 if not high_vram: