AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

correct typo Config Mixin

Browse Source
This commit is contained in:
Patrick von Platen
2022-06-07 10:59:46 +00:00
parent 5b1af9ab82
commit ab8e5364cf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/models/unet.py
View File

@@ -29,7 +29,7 @@ from torchvision import transforms, utils
from PIL import Image
from tqdm import tqdm
from ..configuration_utils import ConfigMixinMixin
from ..configuration_utils import ConfigMixin
from ..modeling_utils import ModelMixin
@@ -175,7 +175,7 @@ class AttnBlock(nn.Module):
return x + h_
class UNetModel(ModelMixin, ConfigMixinMixin):
class UNetModel(ModelMixin, ConfigMixin):
def __init__(
self,
ch=128,