Spaces:
Running
on
Zero
Running
on
Zero
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
|