KingNish commited on
Commit
ef0cbae
1 Parent(s): e926f5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -5,9 +5,7 @@ from diffusers import DiffusionPipeline
5
  from diffusers import StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
6
  import torch
7
 
8
- pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16, variant="fp16").to("cuda")
9
-
10
- pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
11
 
12
  @spaces.GPU(duration=50)
13
  def generate_image(prompt, negative_prompt):
 
5
  from diffusers import StableDiffusionXLPipeline, DPMSolverSinglestepScheduler
6
  import torch
7
 
8
+ pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash").to("cuda")
 
 
9
 
10
  @spaces.GPU(duration=50)
11
  def generate_image(prompt, negative_prompt):