Spaces:
Running
on
A10G
Running
on
A10G
Linoy Tsaban
commited on
Commit
•
57b4fef
1
Parent(s):
9a6ef24
Update scheduling_dpmsolver_multistep_inject.py
Browse files
scheduling_dpmsolver_multistep_inject.py
CHANGED
|
@@ -21,8 +21,8 @@ import numpy as np
|
|
| 21 |
import torch
|
| 22 |
|
| 23 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
| 24 |
-
from diffusers.utils import randn_tensor
|
| 25 |
-
|
| 26 |
from diffusers.schedulers.scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SchedulerOutput
|
| 27 |
|
| 28 |
|
|
|
|
| 21 |
import torch
|
| 22 |
|
| 23 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
| 24 |
+
# from diffusers.utils import randn_tensor
|
| 25 |
+
from diffusers.utils.torch_utils import randn_tensor
|
| 26 |
from diffusers.schedulers.scheduling_utils import KarrasDiffusionSchedulers, SchedulerMixin, SchedulerOutput
|
| 27 |
|
| 28 |
|