sayshara commited on
Commit
4dea744
·
1 Parent(s): f1659ba

Back to DDIM Scheduler

Browse files
Files changed (1) hide show
  1. diffqrcoder_wrapper.py +2 -2
diffqrcoder_wrapper.py CHANGED
@@ -1,6 +1,6 @@
1
  # diffqrcoder_wrapper.py
2
  import torch
3
- from diffusers import ControlNetModel, DPMSolverMultistepScheduler
4
  from PIL import Image
5
  import qrcode
6
  from huggingface_hub import hf_hub_download
@@ -71,7 +71,7 @@ def load_pipeline():
71
  use_auth_token=True, # uses the Space's HF token
72
  )
73
 
74
- pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
75
 
76
  # Memory helpers – cheaper attention
77
  try:
 
1
  # diffqrcoder_wrapper.py
2
  import torch
3
+ from diffusers import ControlNetModel, DDIMScheduler
4
  from PIL import Image
5
  import qrcode
6
  from huggingface_hub import hf_hub_download
 
71
  use_auth_token=True, # uses the Space's HF token
72
  )
73
 
74
+ pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
75
 
76
  # Memory helpers – cheaper attention
77
  try: