Spaces:
Build error
Build error
Upload folder using huggingface_hub
Browse files
README.md
CHANGED
|
@@ -2,7 +2,7 @@
|
|
| 2 |
title: ToDo
|
| 3 |
app_file: app.py
|
| 4 |
sdk: gradio
|
| 5 |
-
sdk_version:
|
| 6 |
---
|
| 7 |
# ImprovedTokenMerge
|
| 8 |

|
|
|
|
| 2 |
title: ToDo
|
| 3 |
app_file: app.py
|
| 4 |
sdk: gradio
|
| 5 |
+
sdk_version: 4.19.2
|
| 6 |
---
|
| 7 |
# ImprovedTokenMerge
|
| 8 |

|
app.py
CHANGED
|
@@ -9,7 +9,6 @@ import numpy as np
|
|
| 9 |
from PIL import Image
|
| 10 |
|
| 11 |
pipe = diffusers.StableDiffusionPipeline.from_pretrained("Lykon/DreamShaper").to("cuda", torch.float16)
|
| 12 |
-
pipe.enable_xformers_memory_efficient_attention()
|
| 13 |
pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 14 |
pipe.safety_checker = None
|
| 15 |
|
|
@@ -40,8 +39,6 @@ with gr.Blocks() as demo:
|
|
| 40 |
@spaces.GPU
|
| 41 |
def generate(prompt, seed, steps, height_width, negative_prompt, guidance_scale, method):
|
| 42 |
|
| 43 |
-
pipe.enable_xformers_memory_efficient_attention()
|
| 44 |
-
|
| 45 |
downsample_factor = 2
|
| 46 |
ratio = 0.38
|
| 47 |
merge_method = "downsample" if method == "todo" else "similarity"
|
|
|
|
| 9 |
from PIL import Image
|
| 10 |
|
| 11 |
pipe = diffusers.StableDiffusionPipeline.from_pretrained("Lykon/DreamShaper").to("cuda", torch.float16)
|
|
|
|
| 12 |
pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(pipe.scheduler.config)
|
| 13 |
pipe.safety_checker = None
|
| 14 |
|
|
|
|
| 39 |
@spaces.GPU
|
| 40 |
def generate(prompt, seed, steps, height_width, negative_prompt, guidance_scale, method):
|
| 41 |
|
|
|
|
|
|
|
| 42 |
downsample_factor = 2
|
| 43 |
ratio = 0.38
|
| 44 |
merge_method = "downsample" if method == "todo" else "similarity"
|