Update Model/TRCaptionNet.py

Model/TRCaptionNet.py

@@ -104,7 +104,7 @@ class TRCaptionNetpp(nn.Module):
     @torch.no_grad()
     def generate(self, images, max_length: int = None, min_length: int = 12, num_beams: int = 3,
                  repetition_penalty: float = 1.1):
-        image_embeds = self.vision_encoder(images.half()).float()
+        image_embeds = self.vision_encoder(images).float()
 
         if self.proj is not None:
             image_embeds = self.proj(image_embeds)