LLM/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-04-22 15:41:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

simplify mixins

Browse Source
This commit is contained in:
Kashif Rasul
2026-02-16 12:52:20 +00:00
parent fc5295951a
commit a4fc9f64b2
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/diffusers/models/autoencoders/autoencoder_rae.py
View File

@@ -22,8 +22,6 @@ import torch.nn as nn
import torch.nn.functional as F
from ...configuration_utils import ConfigMixin, register_to_config
from ...loaders import PeftAdapterMixin
from ...loaders.single_file_model import FromOriginalModelMixin
from ...utils import BaseOutput, logging
from ...utils.accelerate_utils import apply_forward_hook
from ..activations import get_activation
@@ -400,9 +398,7 @@ class RAEDecoder(nn.Module):
return RAEDecoderOutput(logits=logits)
class AutoencoderRAE(
ModelMixin, AttentionMixin, AutoencoderMixin, ConfigMixin, FromOriginalModelMixin, PeftAdapterMixin
):
class AutoencoderRAE(ModelMixin, AttentionMixin, AutoencoderMixin, ConfigMixin):
r"""
Representation Autoencoder (RAE) model for encoding images to latents and decoding latents to images.