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"
|