fix config save bug
Browse files
neuraltexture_controlnet.py
CHANGED
@@ -1,7 +1,9 @@
|
|
1 |
from typing import Optional, Tuple, Union
|
2 |
|
3 |
import torch
|
4 |
-
from diffusers.configuration_utils import register_to_config
|
|
|
|
|
5 |
from diffusers.models.controlnet import ControlNetModel, zero_module
|
6 |
from diffusers.models.embeddings import (
|
7 |
TextImageProjection,
|
@@ -193,7 +195,10 @@ class NeuralTextureControlNetModel(ControlNetModel):
|
|
193 |
addition_embed_type_num_heads: int = 64,
|
194 |
shading_hint_channels: int = 12,
|
195 |
):
|
196 |
-
|
|
|
|
|
|
|
197 |
|
198 |
num_attention_heads = num_attention_heads or attention_head_dim
|
199 |
|
|
|
1 |
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 (
|
9 |
TextImageProjection,
|
|
|
195 |
addition_embed_type_num_heads: int = 64,
|
196 |
shading_hint_channels: int = 12,
|
197 |
):
|
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 |
|