setting same dtype in enhanced garment net

src/enhanced_garment_net.py

@@ -100,6 +100,11 @@ class EnhancedGarmentNetWithTimestep(nn.Module):
         ])
 
     def forward(self, x, t, text_embeds):
+        # Ensure all inputs are of the same dtype
+        x = x.to(dtype=self.garment_net.initial[0].weight.dtype)
+        t = t.to(dtype=self.garment_net.initial[0].weight.dtype)
+        text_embeds = text_embeds.to(dtype=self.garment_net.initial[0].weight.dtype)
+        
         # Get garment features
         garment_out, garment_features = self.garment_net(x)
         

src/tryon_pipeline.py

@@ -401,7 +401,10 @@ class StableDiffusionXLInpaintPipeline(
         force_zeros_for_empty_prompt: bool = True,
     ):
         super().__init__()
-        self.garment_net = EnhancedGarmentNetWithTimestep()
+        #self.garment_net = EnhancedGarmentNetWithTimestep()
+        self.garment_net = EnhancedGarmentNetWithTimestep().to(device=self._execution_device, dtype=self.unet.dtype)
+
+
 
         self.register_modules(
             vae=vae,