Spaces:
Sleeping
Sleeping
Back to DDIM Scheduler
Browse files- 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,
|
| 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 =
|
| 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:
|