From 3d180e9eb6e27eb630647db051733435396be6bb Mon Sep 17 00:00:00 2001
From: lllyasviel <lyuminzhang@outlook.com>
Date: Sat, 11 Nov 2023 09:47:54 -0800
Subject: [PATCH] inpaint engine v2.6

---
 fooocus_version.py |  2 +-
 modules/config.py  | 10 +++++++++-
 update_log.md      |  5 +++++
 webui.py           |  2 +-
 4 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/fooocus_version.py b/fooocus_version.py
index ccd04c41..08c641a8 100644
--- a/fooocus_version.py
+++ b/fooocus_version.py
@@ -1 +1 @@
-version = '2.1.787'
+version = '2.1.788'
diff --git a/modules/config.py b/modules/config.py
index b0962cce..1166a2ec 100644
--- a/modules/config.py
+++ b/modules/config.py
@@ -280,7 +280,7 @@ def update_all_model_names():
 
 
 def downloading_inpaint_models(v):
-    assert v in ['v1', 'v2.5']
+    assert v in ['v1', 'v2.5', 'v2.6']
 
     load_file_from_url(
         url='https://huggingface.co/lllyasviel/fooocus_inpaint/resolve/main/fooocus_inpaint_head.pth',
@@ -306,6 +306,14 @@ def downloading_inpaint_models(v):
         )
         patch_file = os.path.join(path_inpaint, 'inpaint_v25.fooocus.patch')
 
+    if v == 'v2.6':
+        load_file_from_url(
+            url='https://huggingface.co/lllyasviel/fooocus_inpaint/resolve/main/inpaint_v26.fooocus.patch',
+            model_dir=path_inpaint,
+            file_name='inpaint_v26.fooocus.patch'
+        )
+        patch_file = os.path.join(path_inpaint, 'inpaint_v26.fooocus.patch')
+
     return head_file, patch_file
 
 
diff --git a/update_log.md b/update_log.md
index 60ed59fd..6e9d2be8 100644
--- a/update_log.md
+++ b/update_log.md
@@ -1,3 +1,8 @@
+# 2.1.788
+
+* Fixed some math problems in previous versions.
+* Inpaint engine v2.6 join the beta test of Fooocus inpaint models. Use it in dev tools -> inpaint engine -> v2.6 .
+
 # 2.1.785
 
 * The `user_path_config.txt` is deprecated since 2.1.785. If you are using it right now, please use the new `config.txt` instead. See also the new documentation in the Readme.
diff --git a/webui.py b/webui.py
index 3acc3624..a2828add 100644
--- a/webui.py
+++ b/webui.py
@@ -300,7 +300,7 @@ with shared.gradio_root:
                                                                minimum=-1, maximum=1.0, step=0.001, value=-1,
                                                                info='Set as negative number to disable. For developer debugging.')
 
-                        inpaint_engine = gr.Dropdown(label='Inpaint Engine', value='v1', choices=['v1', 'v2.5'],
+                        inpaint_engine = gr.Dropdown(label='Inpaint Engine', value='v1', choices=['v1', 'v2.5', 'v2.6'],
                                                      info='Version of Fooocus inpaint model')
 
                     with gr.Tab(label='Control Debug'):