add adaptive model

src/pipeline.py

@@ -58,7 +58,7 @@ class FashionPipeline:
             target_image_size=resolution,
         )
 
-        segm_mask = self.color_segmentation_mask(segm_mask)
+        segm_mask = self.color_segmentation_mask(segm_mask).resize(control_mask.size)
 
         # generate image
         generator = torch.manual_seed(0)
@@ -95,7 +95,7 @@ class FashionPipeline:
         target_image_size: int = 512,
         max_image_size: int = 768,
         divisible: int = 64,
-    ):
+    ) -> Image:
 
         assert target_image_size % divisible == 0
         assert max_image_size % divisible == 0