declare-lab/tango2
92
1a
2 ��'dL � @ s� d dl mZmZ G dd� ded�ZG dd� ded�ZG dd� ded�ZG d d3� d4ed�ZG d d � d ed�ZG d
d� ded�ZG dd� ded�Z G dd� ded�Z5G dd� ded�Z G dd� ded�Z G dd� ded�Z
G dd� ded�ZG dd � d ed�Zd S ) � )� DummyObject�requires_backendsc @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxControlNetModel�flaxc O s t | dg� d S �Nr �r ��self�args�kwargs� r �W/home/deep/Projects/audio_diffusion/diffusers/src/diffusers/utils/dummy_flax_objects.py�__init__ s z FlaxControlNetModel.__init__c O s t | dg� d S r r ��clsr6 r r r r
� from_config s z FlaxControlNetModel.from_configc O s t | dg� d S r r r r r r
�from_pretrained s z#FlaxControlNetModel.from_pretrainedN��__name__�7__module__� __qualname__� _backendsr � classmethodr r r r r r
r s 8r )� metaclassc @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxModelMixinr c O s t | dg� d S r r r r r r
r s zFlaxModelMixin.__init__c O s t | dg� d S r r r r r r
r s zFlaxModelMixin.from_configc O s t | dg� d S r r r r r r
r s z FlaxModelMixin.from_pretrainedNr r r r r
r s 9r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxUNet2DConditionModelr c O s t | dg� d S r r r r r r
r & s z!FlaxUNet2DConditionModel.__init__c O s t | dg� d S r r r r r r
r ) s z$FlaxUNet2DConditionModel.from_configc O s t | dg� d S r r r r r r
r - s z(FlaxUNet2DConditionModel.from_pretrainedNr r r r r
r # s 10r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxAutoencoderKLr c O s t | dg� d S r r r r r r
r 5 s zFlaxAutoencoderKL.__init__c O s t | dg� d S r r r r r r
r 8 s z FlaxAutoencoderKL.from_configc O s t | dg� d S r r r r r r
r < s z!FlaxAutoencoderKL.from_pretrainedNr r r r r
r 2 s 11r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxDiffusionPipeliner c O s t | dg� d S r r r r r r
r D s z FlaxDiffusionPipeline.__init__c O s t | dg� d S r r r r r r
r G s z!FlaxDiffusionPipeline.from_configc O s t | dg� d S r r r r r r
r K s z%FlaxDiffusionPipeline.from_pretrainedNr r r r r
r A s 12r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxDDIMSchedulerr c O s t | dg� d S r r r r r r
r S s zFlaxDDIMScheduler.__init__c O s t | dg� d S r r r r r r
r V s z FlaxDDIMScheduler.from_configc O s t | dg� d S r r r r r r
r Z s z!FlaxDDIMScheduler.from_pretrainedNr r r r r
r P s 13r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxDDPMSchedulerr c O s t | dg� d S r r r r r r
r b s zFlaxDDPMScheduler.__init__c O s t | dg� d S r r r r r r
r e s z FlaxDDPMScheduler.from_configc O s t | dg� d S r r r r r r
r i s z!FlaxDDPMScheduler.from_pretrainedNr r r r r
r _ s 14r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) � FlaxDPMSolverMultistepSchedulerr c O s t | dg� d S r r r r r r
r q s z(FlaxDPMSolverMultistepScheduler.__init__c O s t | dg� d S r r r r r r
r t s z+FlaxDPMSolverMultistepScheduler.from_configc O s t | dg� d S r r r r r r
r x s z/FlaxDPMSolverMultistepScheduler.from_pretrainedNr r r r r
r n s 15r c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxKarrasVeSchedulerr c O s t | dg� d S r r r r r r
r � s z FlaxKarrasVeScheduler.__init__c O s t | dg� d S r r r r r r
r � s z!FlaxKarrasVeScheduler.from_configc O s t | dg� d S r r r r r r
r � s z%FlaxKarrasVeScheduler.from_pretrainedNr r r r r
r! } s 16r! c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxLMSDiscreteSchedulerr c O s t | dg� d S r r r r r r
r � s z!FlaxLMSDiscreteScheduler.__init__c O s t | dg� d S r r r r r r
r � s z$FlaxLMSDiscreteScheduler.from_configc O s t | dg� d S r r r r r r
r � s z(FlaxLMSDiscreteScheduler.from_pretrainedNr r r r r
r" � s 17r" c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxPNDMSchedulerr c O s t | dg� d S r r r r r r
r � s zFlaxPNDMScheduler.__init__c O s t | dg� d S r r r r r r
r � s z FlaxPNDMScheduler.from_configc O s t | dg� d S r r r r r r
r � s z!FlaxPNDMScheduler.from_pretrainedNr r r r r
r# � s 18r# c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxSchedulerMixinr c O s t | dg� d S r r r r r r
r � s zFlaxSchedulerMixin.__init__c O s t | dg� d S r r r r r r
r � s z FlaxSchedulerMixin.from_configc O s t | dg� d S r r r r r r
r � s z"FlaxSchedulerMixin.from_pretrainedNr r r r r
r$ � s 19r$ c @ s2 e Zd ZdgZdd� Zedd� �Zedd� �ZdS ) �FlaxScoreSdeVeSchedulerr c O s t | dg� d S r r r r r r
r � s z FlaxScoreSdeVeScheduler.__init__c O s t | dg� d S r r r r r r
r � s z#FlaxScoreSdeVeScheduler.from_configc O s t | dg� d S r r r r r r
r � s z'FlaxScoreSdeVeScheduler.from_pretrainedNr r r r r
r% � s 20r% N)�utilsr r r r r r r r r r r! r" r# r$ r% r r r r
�<module> s