suvadityamuk
/

StableDiffusionComparison

Model card Files Files and versions

suvadityamuk commited on Dec 7, 2022

Commit

22da2dc

·

1 Parent(s): 706d9f4

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -55,7 +55,7 @@ class StableDiffusionComparisonPipeline(DiffusionPipeline):
         feature_extractor: CLIPFeatureExtractor,
         requires_safety_checker: bool = True,
     ):
-        super()._init_()
         self.pipe1 = StableDiffusionPipeline.from_pretrained(pipe1_model_id)
         self.pipe2 = StableDiffusionPipeline.from_pretrained(pipe2_model_id)

         feature_extractor: CLIPFeatureExtractor,
         requires_safety_checker: bool = True,
     ):
+        super().__init__()
         self.pipe1 = StableDiffusionPipeline.from_pretrained(pipe1_model_id)
         self.pipe2 = StableDiffusionPipeline.from_pretrained(pipe2_model_id)