NCJ commited on
Commit
a5ccd0f
1 Parent(s): db6ca6b

upgrade to diffusers 0.29

Browse files
Files changed (1) hide show
  1. neuraltexture_controlnet.py +2 -2
neuraltexture_controlnet.py CHANGED
@@ -2,7 +2,7 @@ from typing import Optional, Tuple, Union
2
 
3
  import torch
4
  from diffusers.configuration_utils import ConfigMixin, register_to_config
5
- from diffusers.loaders.controlnet import FromOriginalControlNetMixin
6
  from diffusers.models.modeling_utils import ModelMixin
7
  from diffusers.models.controlnet import ControlNetModel, zero_module
8
  from diffusers.models.embeddings import (
@@ -198,7 +198,7 @@ class NeuralTextureControlNetModel(ControlNetModel):
198
  # avoid running __init__() of the original ControlNetModel
199
  super(ModelMixin, self).__init__()
200
  super(ConfigMixin, self).__init__()
201
- super(FromOriginalControlNetMixin, self).__init__()
202
 
203
  num_attention_heads = num_attention_heads or attention_head_dim
204
 
 
2
 
3
  import torch
4
  from diffusers.configuration_utils import ConfigMixin, register_to_config
5
+ from diffusers.loaders.single_file_model import FromOriginalModelMixin
6
  from diffusers.models.modeling_utils import ModelMixin
7
  from diffusers.models.controlnet import ControlNetModel, zero_module
8
  from diffusers.models.embeddings import (
 
198
  # avoid running __init__() of the original ControlNetModel
199
  super(ModelMixin, self).__init__()
200
  super(ConfigMixin, self).__init__()
201
+ super(FromOriginalModelMixin, self).__init__()
202
 
203
  num_attention_heads = num_attention_heads or attention_head_dim
204