reverting enhanced garment net

src/tryon_pipeline.py

@@ -400,7 +400,7 @@ class StableDiffusionXLInpaintPipeline(
         force_zeros_for_empty_prompt: bool = True,
     ):
         super().__init__()
-        self.garment_net = EnhancedGarmentNetWithTimestep()
+        # self.garment_net = EnhancedGarmentNetWithTimestep()
 
         self.register_modules(
             vae=vae,
@@ -1783,11 +1783,12 @@ class StableDiffusionXLInpaintPipeline(
                 added_cond_kwargs = {"text_embeds": add_text_embeds, "time_ids": add_time_ids}
                 if ip_adapter_image is not None:
                     added_cond_kwargs["image_embeds"] = image_embeds
+                print("Calling unet encoder for garment feature extraction")
                 # down,reference_features = self.UNet_Encoder(cloth,t, text_embeds_cloth,added_cond_kwargs= {"text_embeds": pooled_prompt_embeds_c, "time_ids": add_time_ids},return_dict=False)
-                # down,reference_features = self.unet_encoder(cloth,t, text_embeds_cloth,return_dict=False)
-                garment_out, reference_features = self.garment_net(cloth, t, text_embeds_cloth)
-                # print(type(reference_features))
-                # print(reference_features)
+                down,reference_features = self.unet_encoder(cloth,t, text_embeds_cloth,return_dict=False)                
+                #garment_out, reference_features = self.garment_net(cloth, t, text_embeds_cloth)
+                print(type(reference_features))
+                print(reference_features)
                 reference_features = list(reference_features)
                 # print(len(reference_features))
                 # for elem in reference_features: