Unverified Commit 27d11a00 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[K-LMS Scheduler] fix import (#191)

parent 554e67cb
...@@ -31,13 +31,13 @@ from .schedulers import ( ...@@ -31,13 +31,13 @@ from .schedulers import (
if is_scipy_available(): if is_scipy_available():
from .schedulers import LMSDiscreteScheduler from .schedulers import LMSDiscreteScheduler
else:
from .utils.dummy_scipy_objects import *
from .training_utils import EMAModel from .training_utils import EMAModel
if is_transformers_available(): if is_transformers_available():
from .pipelines import LDMTextToImagePipeline, StableDiffusionPipeline from .pipelines import LDMTextToImagePipeline, StableDiffusionPipeline
else: else:
from .utils.dummy_transformers_objects import * from .utils.dummy_transformers_objects import *
...@@ -28,3 +28,5 @@ from .scheduling_utils import SchedulerMixin ...@@ -28,3 +28,5 @@ from .scheduling_utils import SchedulerMixin
if is_scipy_available(): if is_scipy_available():
from .scheduling_lms_discrete import LMSDiscreteScheduler from .scheduling_lms_discrete import LMSDiscreteScheduler
else:
from ..utils.dummy_scipy_objects import *
...@@ -120,6 +120,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin): ...@@ -120,6 +120,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
self.timesteps = self.prk_timesteps + self.plms_timesteps self.timesteps = self.prk_timesteps + self.plms_timesteps
self.ets = []
self.counter = 0 self.counter = 0
self.set_format(tensor_format=self.tensor_format) self.set_format(tensor_format=self.tensor_format)
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
from ..utils import DummyObject, requires_backends from ..utils import DummyObject, requires_backends
class LmsDiscreteScheduler(metaclass=DummyObject): class LMSDiscreteScheduler(metaclass=DummyObject):
_backends = ["scipy"] _backends = ["scipy"]
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment