Spaces:
Sleeping
Sleeping
import ModelMixin properly from latest diffusers
Browse files
latentsync/models/unet.py
CHANGED
|
@@ -9,7 +9,7 @@ import torch.nn as nn
|
|
| 9 |
import torch.utils.checkpoint
|
| 10 |
|
| 11 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
| 12 |
-
from diffusers.modeling_utils import ModelMixin
|
| 13 |
from diffusers import UNet2DConditionModel
|
| 14 |
from diffusers.utils import BaseOutput, logging
|
| 15 |
from diffusers.models.embeddings import TimestepEmbedding, Timesteps
|
|
|
|
| 9 |
import torch.utils.checkpoint
|
| 10 |
|
| 11 |
from diffusers.configuration_utils import ConfigMixin, register_to_config
|
| 12 |
+
from diffusers.models.modeling_utils import ModelMixin
|
| 13 |
from diffusers import UNet2DConditionModel
|
| 14 |
from diffusers.utils import BaseOutput, logging
|
| 15 |
from diffusers.models.embeddings import TimestepEmbedding, Timesteps
|