Astridkraft commited on
Commit
1a7921e
·
verified ·
1 Parent(s): ecfe07b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -48,7 +48,12 @@ def load_img2img():
48
 
49
  # DPMSolver für Image-to-Image
50
  from diffusers import DPMSolverMultistepScheduler
51
- pipe_img2img.scheduler = DPMSolverMultistepScheduler.from_config(pipe_img2img.scheduler.config)
 
 
 
 
 
52
 
53
  pipe_img2img.enable_attention_slicing()
54
  return pipe_img2img
 
48
 
49
  # DPMSolver für Image-to-Image
50
  from diffusers import DPMSolverMultistepScheduler
51
+ pipe_img2img.scheduler = DPMSolverMultistepScheduler.from_config(pipe_img2img.scheduler.config)(
52
+ pipe_img2img.scheduler.config,
53
+ algorithm_type="sde-dpmsolver++", # ← Hier ändern!
54
+ use_karras_sigmas=True,
55
+ timestep_spacing="linspace"
56
+ )
57
 
58
  pipe_img2img.enable_attention_slicing()
59
  return pipe_img2img