Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "1f0705adcfe9e57db1bb90c2938922249b186cfc"
Commit ab8e5364 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

correct typo Config Mixin

parent 5b1af9ab
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/models/unet.py
View file @ ab8e5364
...@@ -29,7 +29,7 @@ from torchvision import transforms, utils ...@@ -29,7 +29,7 @@ from torchvision import transforms, utils
from PIL import Image from PIL import Image
from tqdm import tqdm from tqdm import tqdm
from ..configuration_utils import ConfigMixinMixin from ..configuration_utils import ConfigMixin
from ..modeling_utils import ModelMixin from ..modeling_utils import ModelMixin
...@@ -175,7 +175,7 @@ class AttnBlock(nn.Module): ...@@ -175,7 +175,7 @@ class AttnBlock(nn.Module):
return x + h_ return x + h_
class UNetModel(ModelMixin, ConfigMixinMixin): class UNetModel(ModelMixin, ConfigMixin):
def __init__( def __init__(
self, self,
ch=128, ch=128,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment