Update conditioner.py

2026-01-23 20:54:31 +08:00 · 2025-02-03 19:49:18 +02:00 · 2025-02-03 19:49:18 +02:00 · d618646a88
commit d618646a88
parent df4e4f7337
1 changed files with 1 additions and 1 deletions
--- a/hy3dgen/shapegen/models/conditioner.py
+++ b/hy3dgen/shapegen/models/conditioner.py
@ -78,7 +78,7 @@ class ImageEncoder(nn.Module):
            mask = mask.to(image)
            image = image * mask

-        inputs = image
+        inputs = self.transform(image)
        outputs = self.model(inputs)

        last_hidden_state = outputs.last_hidden_state