From 3121d5ec35bea42d1498a4d787fb012ac8ed7b9b Mon Sep 17 00:00:00 2001
From: bmaltais <bernard@ducourier.com>
Date: Fri, 2 Aug 2024 10:41:13 -0400
Subject: [PATCH] Fix issue with missing key. Upgrade Gradio release for
 security issue.

---
 .release                | 2 +-
 kohya_gui/common_gui.py | 1 +
 requirements.txt        | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.release b/.release
index 3f16892e3..7d901a22e 100644
--- a/.release
+++ b/.release
@@ -1 +1 @@
-v24.1.5
\ No newline at end of file
+v24.1.6
\ No newline at end of file
diff --git a/kohya_gui/common_gui.py b/kohya_gui/common_gui.py
index 8fe216bd5..0ca334eb6 100644
--- a/kohya_gui/common_gui.py
+++ b/kohya_gui/common_gui.py
@@ -358,6 +358,7 @@ def update_my_data(my_data):
 
     for key in [
         "max_train_steps",
+        "caption_dropout_every_n_epochs"
     ]:
         value = my_data.get(key)
         if value is not None:
diff --git a/requirements.txt b/requirements.txt
index 3d34ae8f2..73506471e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,7 +7,7 @@ easygui==0.98.3
 einops==0.7.0
 fairscale==0.4.13
 ftfy==6.1.1
-gradio==4.36.0
+gradio==4.36.1
 huggingface-hub==0.20.1
 imagesize==1.4.1
 invisible-watermark==0.2.0