Skip to content

Commit 27d11a0

Browse files
[K-LMS Scheduler] fix import (#191)
1 parent 554e67c commit 27d11a0

File tree

4 files changed

+6
-3
lines changed

4 files changed

+6
-3
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,13 +31,13 @@
3131

3232
if is_scipy_available():
3333
from .schedulers import LMSDiscreteScheduler
34+
else:
35+
from .utils.dummy_scipy_objects import *
3436

3537
from .training_utils import EMAModel
3638

3739

3840
if is_transformers_available():
3941
from .pipelines import LDMTextToImagePipeline, StableDiffusionPipeline
40-
41-
4242
else:
4343
from .utils.dummy_transformers_objects import *

src/diffusers/schedulers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,3 +28,5 @@
2828

2929
if is_scipy_available():
3030
from .scheduling_lms_discrete import LMSDiscreteScheduler
31+
else:
32+
from ..utils.dummy_scipy_objects import *

src/diffusers/schedulers/scheduling_pndm.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,7 @@ def set_timesteps(self, num_inference_steps, offset=0):
120120

121121
self.timesteps = self.prk_timesteps + self.plms_timesteps
122122

123+
self.ets = []
123124
self.counter = 0
124125
self.set_format(tensor_format=self.tensor_format)
125126

src/diffusers/utils/dummy_scipy_objects.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
from ..utils import DummyObject, requires_backends
44

55

6-
class LmsDiscreteScheduler(metaclass=DummyObject):
6+
class LMSDiscreteScheduler(metaclass=DummyObject):
77
_backends = ["scipy"]
88

99
def __init__(self, *args, **kwargs):

0 commit comments

Comments
 (0)