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
|