declare-lab/tango2
92
1a
2 ��'d�; � @ sx 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�ZG d d � d ed�ZG d d � d ed�ZG d!d"� d"ed�ZG d#d$� d$ed�ZG d%d&� d&ed�ZG d'd(� d(ed�ZG d)d*� d*ed�ZG d+d,� d,ed�ZG d-d.� d.ed�ZG d/d0� d0ed�ZG d1d2� d2ed�ZG d3d4� d4ed�ZG d5d6� d6ed�Z G d7d8� d8ed�Z G d9d:� d:ed�Z G d;d<� d<ed�Z G d=d>� d>ed�Z G d?d@� d@ed�Z!G dAdB� dBed�Z"G dCdD� dDed�Z#G dEdF� dFed�Z$dGS )H� )� DummyObject�requires_backendsc @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )6�TextualInversionLoaderMixin�torch� transformersc O s t | ddg� d S �Nr r �r ��self�args�kwargs� r
�i/home/deep/Projects/audio_diffusion/diffusers/src/diffusers/utils/dummy_torch_and_transformers_objects.py�__init__ s z$TextualInversionLoaderMixin.__init__c O s t | ddg� d S r r ��clsr r r
r
r � from_config s z'TextualInversionLoaderMixin.from_configc O s t | ddg� d S r r r r
r
r �from_pretrained s z+TextualInversionLoaderMixin.from_pretrainedN��__name__�7__module__� __qualname__� _backendsr � classmethodr r r
r
r
r r s 8r )� metaclassc @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )9�AltDiffusionImg2ImgPipeliner r c O s t | ddg� d S r r r r
r
r r s z$AltDiffusionImg2ImgPipeline.__init__c O s t | ddg� d S r r r r
r
r r s z'AltDiffusionImg2ImgPipeline.from_configc O s t | ddg� d S r r r r
r
r r s z+AltDiffusionImg2ImgPipeline.from_pretrainedNr r
r
r
r r s 10r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )11�AltDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r & s z AltDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r ) s z AltDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r - s z$AltDiffusionPipeline.from_pretrainedNr r
r
r
r r # s 12r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )13�AudioLDMPipeliner r c O s t | ddg� d S r r r r
r
r r 5 s zAudioLDMPipeline.__init__c O s t | ddg� d S r r r r
r
r r 8 s z AudioLDMPipeline.from_configc O s t | ddg� d S r r r r
r
r r < s z AudioLDMPipeline.from_pretrainedNr r
r
r
r r 2 s 14r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )15�CycleDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r D s z CycleDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r G s z"CycleDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r K s z&CycleDiffusionPipeline.from_pretrainedNr r
r
r
r r A s 16r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )17�LDMTextToImagePipeliner r c O s t | ddg� d S r r r r
r
r r S s z LDMTextToImagePipeline.__init__c O s t | ddg� d S r r r r
r
r r V s z"LDMTextToImagePipeline.from_configc O s t | ddg� d S r r r r
r
r r Z s z&LDMTextToImagePipeline.from_pretrainedNr r
r
r
r r P s 18r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )19�PaintByExamplePipeliner r c O s t | ddg� d S r r r r
r
r r b s z PaintByExamplePipeline.__init__c O s t | ddg� d S r r r r
r
r r e s z"PaintByExamplePipeline.from_configc O s t | ddg� d S r r r r
r
r r i s z&PaintByExamplePipeline.from_pretrainedNr r
r
r
r r _ s 20r c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )21� SemanticStableDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r q s z(SemanticStableDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r t s z+SemanticStableDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r x s z/SemanticStableDiffusionPipeline.from_pretrainedNr r
r
r
r r! n s 22r! c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )23�&StableDiffusionAttendAndExcitePipeliner r c O s t | ddg� d S r r r r
r
r r � s z/StableDiffusionAttendAndExcitePipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z2StableDiffusionAttendAndExcitePipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z6StableDiffusionAttendAndExcitePipeline.from_pretrainedNr r
r
r
r r" } s 24r" c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )25�!StableDiffusionControlNetPipeliner r c O s t | ddg� d S r r r r
r
r r � s z*StableDiffusionControlNetPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z-StableDiffusionControlNetPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z1StableDiffusionControlNetPipeline.from_pretrainedNr r
r
r
r r# � s 26r# c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )27� StableDiffusionDepth2ImgPipeliner r c O s t | ddg� d S r r r r
r
r r � s z)StableDiffusionDepth2ImgPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z,StableDiffusionDepth2ImgPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z0StableDiffusionDepth2ImgPipeline.from_pretrainedNr r
r
r
r r$ � s 28r$ c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )29�%StableDiffusionImageVariationPipeliner r c O s t | ddg� d S r r r r
r
r r � s z.StableDiffusionImageVariationPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z1StableDiffusionImageVariationPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z5StableDiffusionImageVariationPipeline.from_pretrainedNr r
r
r
r r% � s 30r% c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )31� StableDiffusionImg2ImgPipeliner r c O s t | ddg� d S r r r r
r
r r � s z'StableDiffusionImg2ImgPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z*StableDiffusionImg2ImgPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z.StableDiffusionImg2ImgPipeline.from_pretrainedNr r
r
r
r r& � s 32r& c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )33� StableDiffusionInpaintPipeliner r c O s t | ddg� d S r r r r
r
r r � s z'StableDiffusionInpaintPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z*StableDiffusionInpaintPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z.StableDiffusionInpaintPipeline.from_pretrainedNr r
r
r
r r' � s 34r' c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )35�$StableDiffusionInpaintPipelineLegacyr r c O s t | ddg� d S r r r r
r
r r � s z-StableDiffusionInpaintPipelineLegacy.__init__c O s t | ddg� d S r r r r
r
r r � s z0StableDiffusionInpaintPipelineLegacy.from_configc O s t | ddg� d S r r r r
r
r r � s z4StableDiffusionInpaintPipelineLegacy.from_pretrainedNr r
r
r
r r( � s 36r( c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )37�&StableDiffusionInstructPix2PixPipeliner r c O s t | ddg� d S r r r r
r
r r � s z/StableDiffusionInstructPix2PixPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z2StableDiffusionInstructPix2PixPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z6StableDiffusionInstructPix2PixPipeline.from_pretrainedNr r
r
r
r r) � s 38r) c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )39�$StableDiffusionLatentUpscalePipeliner r c O s t | ddg� d S r r r r
r
r r � s z-StableDiffusionLatentUpscalePipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z0StableDiffusionLatentUpscalePipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z4StableDiffusionLatentUpscalePipeline.from_pretrainedNr r
r
r
r r* � s 40r* c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )41�#StableDiffusionModelEditingPipeliner r c O s t | ddg� d S r r r r
r
r r s z,StableDiffusionModelEditingPipeline.__init__c O s t | ddg� d S r r r r
r
r r 42 s z/StableDiffusionModelEditingPipeline.from_configc O s t | ddg� d S r r r r
r
r r s z3StableDiffusionModelEditingPipeline.from_pretrainedNr r
r
r
r r+ s 43r+ c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )44� StableDiffusionPanoramaPipeliner r c O s t | ddg� d S r r r r
r
r r s z(StableDiffusionPanoramaPipeline.__init__c O s t | ddg� d S r r r r
r
r r s z+StableDiffusionPanoramaPipeline.from_configc O s t | ddg� d S r r r r
r
r r s z/StableDiffusionPanoramaPipeline.from_pretrainedNr r
r
r
r r, s 45r, c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )46�StableDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r % s z StableDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r ( s z#StableDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r , s z'StableDiffusionPipeline.from_pretrainedNr r
r
r
r r- " s 47r- c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )48�StableDiffusionPipelineSafer r c O s t | ddg� d S r r r r
r
r r 4 s z$StableDiffusionPipelineSafe.__init__c O s t | ddg� d S r r r r
r
r r 7 s z'StableDiffusionPipelineSafe.from_configc O s t | ddg� d S r r r r
r
r r ; s z+StableDiffusionPipelineSafe.from_pretrainedNr r
r
r
r r. 1 s 49r. c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )50�"StableDiffusionPix2PixZeroPipeliner r c O s t | ddg� d S r r r r
r
r r C s z+StableDiffusionPix2PixZeroPipeline.__init__c O s t | ddg� d S r r r r
r
r r F s z.StableDiffusionPix2PixZeroPipeline.from_configc O s t | ddg� d S r r r r
r
r r J s z2StableDiffusionPix2PixZeroPipeline.from_pretrainedNr r
r
r
r r/ @ s 51r/ c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )52�StableDiffusionSAGPipeliner r c O s t | ddg� d S r r r r
r
r r R s z#StableDiffusionSAGPipeline.__init__c O s t | ddg� d S r r r r
r
r r U s z&StableDiffusionSAGPipeline.from_configc O s t | ddg� d S r r r r
r
r r Y s z*StableDiffusionSAGPipeline.from_pretrainedNr r
r
r
r r0 O s 53r0 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )54� StableDiffusionUpscalePipeliner r c O s t | ddg� d S r r r r
r
r r a s z'StableDiffusionUpscalePipeline.__init__c O s t | ddg� d S r r r r
r
r r d s z*StableDiffusionUpscalePipeline.from_configc O s t | ddg� d S r r r r
r
r r h s z.StableDiffusionUpscalePipeline.from_pretrainedNr r
r
r
r r1 ^ s 55r1 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )56�StableUnCLIPImg2ImgPipeliner r c O s t | ddg� d S r r r r
r
r r p s z$StableUnCLIPImg2ImgPipeline.__init__c O s t | ddg� d S r r r r
r
r r s s z'StableUnCLIPImg2ImgPipeline.from_configc O s t | ddg� d S r r r r
r
r r w s z+StableUnCLIPImg2ImgPipeline.from_pretrainedNr r
r
r
r r2 m s 57r2 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )58�StableUnCLIPPipeliner r c O s t | ddg� d S r r r r
r
r r s z StableUnCLIPPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z StableUnCLIPPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z$StableUnCLIPPipeline.from_pretrainedNr r
r
r
r r3 | s 59r3 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )60�TextToVideoSDPipeliner r c O s t | ddg� d S r r r r
r
r r � s z TextToVideoSDPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z!TextToVideoSDPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z%TextToVideoSDPipeline.from_pretrainedNr r
r
r
r r4 � s 61r4 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )62� UnCLIPImageVariationPipeliner r c O s t | ddg� d S r r r r
r
r r � s z%UnCLIPImageVariationPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z(UnCLIPImageVariationPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z,UnCLIPImageVariationPipeline.from_pretrainedNr r
r
r
r r5 � s 63r5 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )64�UnCLIPPipeliner r c O s t | ddg� d S r r r r
r
r r � s zUnCLIPPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s zUnCLIPPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z UnCLIPPipeline.from_pretrainedNr r
r
r
r r6 � s 65r6 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )66�$VersatileDiffusionDualGuidedPipeliner r c O s t | ddg� d S r r r r
r
r r � s z-VersatileDiffusionDualGuidedPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z0VersatileDiffusionDualGuidedPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z4VersatileDiffusionDualGuidedPipeline.from_pretrainedNr r
r
r
r r7 � s 67r7 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )68�(VersatileDiffusionImageVariationPipeliner r c O s t | ddg� d S r r r r
r
r r � s z1VersatileDiffusionImageVariationPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z4VersatileDiffusionImageVariationPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z8VersatileDiffusionImageVariationPipeline.from_pretrainedNr r
r
r
r r8 � s 69r8 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )70�VersatileDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r � s z#VersatileDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z&VersatileDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z*VersatileDiffusionPipeline.from_pretrainedNr r
r
r
r r9 � s 71r9 c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )72�%VersatileDiffusionTextToImagePipeliner r c O s t | ddg� d S r r r r
r
r r � s z.VersatileDiffusionTextToImagePipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z1VersatileDiffusionTextToImagePipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z5VersatileDiffusionTextToImagePipeline.from_pretrainedNr r
r
r
r r: � s 73r: c @ s4 e Zd ZddgZdd� Zedd� �Zedd� �Zd S )74�VQDiffusionPipeliner r c O s t | ddg� d S r r r r
r
r r � s z VQDiffusionPipeline.__init__c O s t | ddg� d S r r r r
r
r r � s z VQDiffusionPipeline.from_configc O s t | ddg� d S r r r r
r
r r � s z#VQDiffusionPipeline.from_pretrainedNr r
r
r
r r; � s 75r; N)%�utilsr r r r r r r r r r! r" r# r$ r% r& r' r( r) r* r+ r, r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r
r
r
r �<module> sD