Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

correct typo Config Mixin

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
Show Stats 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,