fffiloni commited on
Commit
af1fddf
·
verified ·
1 Parent(s): 8c0a112

import ModelMixin properly from latest diffusers

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