update README, add screenshots, fix minor bugs

seruva19 · Jan 2, 2025 · e7c2562 · e7c2562
1 parent f8891ae
commit e7c2562
Show file tree

Hide file tree

Showing 12 changed files with 21 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -43,7 +43,7 @@ Web-GUI for Kandinsky generative AI models ([Kandinsky2](https://github.com/ai-f
 ## Screenshots
 
 <details>
-    <summary>Toggle</summary>
+<summary>Toggle</summary>
 
 ### Kandinsky 2.0
 
@@ -67,7 +67,7 @@ Web-GUI for Kandinsky generative AI models ([Kandinsky2](https://github.com/ai-f
 
 ### Kandinsky 4.0
 
-![img](/sshots/video_40.png)
+![img](/sshots/screenshot_40.gif)
 
 </details>
 

diff --git a/requirements.txt b/requirements.txt
@@ -3,11 +3,11 @@ setuptools==69.5.1
 git+https://github.com/ai-forever/Kandinsky-2.git@33c3176
 git+https://github.com/openai/CLIP.git@a1d0717
 diffusers==0.30.3
-git+https://github.com/huggingface/transformers@6544271  
+transformers==4.44.0
 torchao==0.7.0
 optimum-quanto==0.2.6
 mmgp==1.2.0
-tokenizers==0.20.1
+tokenizers==0.19.1
 accelerate==1.0.1
 safetensors==0.4.5
 opencv-python==4.8.0.74

diff --git a/src/converter/kd_to_diffusers.py b/src/converter/kd_to_diffusers.py
diff --git a/src/ui_blocks/settings_system.py b/src/ui_blocks/settings_system.py
@@ -41,9 +41,17 @@ def update_info():
     try:
         import flash_attn
 
-        flash_attn_info = f"flash_attn: {flash_attn.__version__}\n"
+        flash_attn_info = f"flashattn: {flash_attn.__version__}\n"
     except:
-        flash_attn_info = f"flash_attn: not installed\n"
+        flash_attn_info = f"flashattn: not installed\n"
+
+    sageattn_info = ""
+    try:
+        from sageattention import sageattn
+
+        sageattn_info = f"sageattn: {sageattention.__version__}\n"
+    except:
+        sageattn_info = f"sageattn: not installed\n"
 
     diffusers_info = ""
     try:

diff --git a/src/ui_blocks/t2i.py b/src/ui_blocks/t2i.py
@@ -349,7 +349,13 @@ async def generate(
                         "init_image": None,
                     }
 
-                    shared.storage.save(block, params)
+                    saved_params = {
+                        k: v
+                        for k, v in params.items()
+                        if k not in ["cnet_image", "init_image"]
+                    }
+                    shared.storage.save(block, saved_params)
+
                     params = augmentations["exec"](params, injections)
 
                     yield generate_fn(params)

diff --git a/sshots/screenshot.png b/sshots/screenshot.png
diff --git a/sshots/screenshot_20.png b/sshots/screenshot_20.png
diff --git a/sshots/screenshot_21.png b/sshots/screenshot_21.png
diff --git a/sshots/screenshot_22.png b/sshots/screenshot_22.png
diff --git a/sshots/screenshot_30.png b/sshots/screenshot_30.png
diff --git a/sshots/screenshot_31.png b/sshots/screenshot_31.png
diff --git a/sshots/screenshot_40.gif b/sshots/screenshot_40.gif