tolgacangoz
/

matryoshka-diffusion-models

Model card Files Files and versions Community

tolgacangoz commited on Oct 8, 2024

Commit

42aafad

·

verified ·

1 Parent(s): f815fa7

Upload matryoshka.py

Files changed (1) hide show

unet/matryoshka.py +1 -1

unet/matryoshka.py CHANGED Viewed

@@ -3762,7 +3762,7 @@ class MatryoshkaPipeline(
         text_encoder: T5EncoderModel,
         tokenizer: T5TokenizerFast,
         scheduler: MatryoshkaDDIMScheduler,
-        unet: MatryoshkaUNet2DConditionModel = None,
         feature_extractor: CLIPImageProcessor = None,
         image_encoder: CLIPVisionModelWithProjection = None,
         trust_remote_code: bool = False,

         text_encoder: T5EncoderModel,
         tokenizer: T5TokenizerFast,
         scheduler: MatryoshkaDDIMScheduler,
+        unet: MatryoshkaUNet2DConditionModel,
         feature_extractor: CLIPImageProcessor = None,
         image_encoder: CLIPVisionModelWithProjection = None,
         trust_remote_code: bool = False,