patrickvonplaten commited on
Commit
61ba2f1
1 Parent(s): 8ae0dea

Update scheduler/my_scheduler.py

Browse files
Files changed (1) hide show
  1. scheduler/my_scheduler.py +1 -1
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