"vscode:/vscode.git/clone" did not exist on "a283ec2eece57454ec9301e5542cffa1201e175f"
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 (
if is_scipy_available():
from .schedulers import LMSDiscreteScheduler
else:
from .utils.dummy_scipy_objects import *
from .training_utils import EMAModel
if is_transformers_available():
from .pipelines import LDMTextToImagePipeline, StableDiffusionPipeline
else:
from .utils.dummy_transformers_objects import *
......@@ -28,3 +28,5 @@ from .scheduling_utils import SchedulerMixin
if is_scipy_available():
from .scheduling_lms_discrete import LMSDiscreteScheduler
else:
from ..utils.dummy_scipy_objects import *
......@@ -120,6 +120,7 @@ class PNDMScheduler(SchedulerMixin, ConfigMixin):
self.timesteps = self.prk_timesteps + self.plms_timesteps
self.ets = []
self.counter = 0
self.set_format(tensor_format=self.tensor_format)
......
......@@ -3,7 +3,7 @@
from ..utils import DummyObject, requires_backends
class LmsDiscreteScheduler(metaclass=DummyObject):
class LMSDiscreteScheduler(metaclass=DummyObject):
_backends = ["scipy"]
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