upgrade to diffusers 0.29
Browse files
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.
|
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(
|
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 |
|