Upload folder using huggingface_hub

README.md

@@ -8,6 +8,7 @@ sdk_version: 5.14.0
 
 # Say hi to LightDiffusion-Next 👋
 
+[![demo platform](https://img.shields.io/badge/Play%20with%20LightDiffusion%21-LightDiffusion%20demo%20platform-lightblue)](https://huggingface.co/spaces/Aatricks/LightDiffusion-Next) 
 
 **LightDiffusion-Next**  is the fastest AI-powered image generation GUI/CLI, combining speed, precision, and flexibility in one cohesive tool.
 </br>
@@ -19,7 +20,6 @@ sdk_version: 5.14.0
 </br>
 </div>
 
-
 As a refactored and improved version of the original [LightDiffusion repository](https://github.com/Aatrick/LightDiffusion), this project enhances usability, maintainability, and functionality while introducing a host of new features to streamline your creative workflows.
 
 ## Motivation:

app.py

@@ -91,6 +91,7 @@ def generate_images(
 with gr.Blocks(title="LightDiffusion Web UI") as demo:
     gr.Markdown("# LightDiffusion Web UI")
     gr.Markdown("Generate AI images using LightDiffusion")
+    gr.Markdown("")
     
     with gr.Row():
         with gr.Column():

modules/Attention/AttentionMethods.py

@@ -6,7 +6,7 @@ import torch
 
 
 def attention_xformers(
-    q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, heads: int, mask=None
+    q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, heads: int, mask=None, skip_reshape=False
 ) -> torch.Tensor:
     """#### Make an attention call using xformers. Fastest attention implementation.
 
@@ -44,7 +44,7 @@ def attention_xformers(
 
 
 def attention_pytorch(
-    q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, heads: int, mask=None
+    q: torch.Tensor, k: torch.Tensor, v: torch.Tensor, heads: int, mask=None, skip_reshape=False
 ) -> torch.Tensor:
     """#### Make an attention call using PyTorch.