patrickvonplaten
commited on
Commit
•
61ba2f1
1
Parent(s):
8ae0dea
Update scheduler/my_scheduler.py
Browse files
scheduler/my_scheduler.py
CHANGED
@@ -24,7 +24,7 @@ import torch
|
|
24 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
25 |
from diffusers.utils import BaseOutput
|
26 |
from diffusers.utils.torch_utils import randn_tensor
|
27 |
-
from diffusers.scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
|
28 |
|
29 |
|
30 |
@dataclass
|
|
|
24 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
25 |
from diffusers.utils import BaseOutput
|
26 |
from diffusers.utils.torch_utils import randn_tensor
|
27 |
+
from diffusers.schedulers.scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin
|
28 |
|
29 |
|
30 |
@dataclass
|