aningineer commited on
Commit
10d5e0b
1 Parent(s): 6b566c6

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +0 -3
README.md CHANGED
@@ -2,7 +2,7 @@
2
  title: ToDo
3
  app_file: app.py
4
  sdk: gradio
5
- sdk_version: 3.50.2
6
  ---
7
  # ImprovedTokenMerge
8
  ![compare3.png](compare3.png)
 
2
  title: ToDo
3
  app_file: app.py
4
  sdk: gradio
5
+ sdk_version: 4.19.2
6
  ---
7
  # ImprovedTokenMerge
8
  ![compare3.png](compare3.png)
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"