Update optimizer.py

optimizer.py

@@ -190,27 +190,23 @@ class UltraSupremeOptimizer:
     def run_clip_inference(self, image: Image.Image) -> Tuple[str, str, str]:
         """Solo la inferencia CLIP usa GPU"""
         try:
-            # Preparar modelos para GPU
-            self._prepare_models_for_gpu()
-            
-            # Usar autocast para manejar precisión mixta
-            with torch.cuda.amp.autocast(enabled=True, dtype=torch.float16):
-                # Convertir imagen a tensor y asegurar que esté en half precision
-                from torchvision import transforms
-                preprocess = transforms.Compose([
-                    transforms.Resize((224, 224)),
-                    transforms.ToTensor(),
-                    transforms.Normalize(mean=[0.48145466, 0.4578275, 0.40821073], 
-                                       std=[0.26862954, 0.26130258, 0.27577711]),
-                ])
+            # NO usar half precision - mantener float32 para compatibilidad
+            if hasattr(self.interrogator, 'caption_model'):
+                self.interrogator.caption_model = self.interrogator.caption_model.to("cuda")
                 
-                # Procesar imagen manualmente para controlar la precisión
-                image_tensor = preprocess(image).unsqueeze(0).half().to("cuda")
+            if hasattr(self.interrogator, 'clip_model'):
+                self.interrogator.clip_model = self.interrogator.clip_model.to("cuda")
                 
-                # Ejecutar inferencias con manejo especial
-                full_prompt = self._safe_interrogate(image, 'interrogate')
-                clip_fast = self._safe_interrogate(image, 'interrogate_fast')
-                clip_classic = self._safe_interrogate(image, 'interrogate_classic')
+            if hasattr(self.interrogator, 'blip_model'):
+                self.interrogator.blip_model = self.interrogator.blip_model.to("cuda")
+                
+            self.interrogator.config.device = "cuda"
+            logger.info("Models moved to GPU with float32 (full precision)")
+            
+            # Ejecutar inferencias sin autocast para evitar problemas de half precision
+            full_prompt = self.interrogator.interrogate(image)
+            clip_fast = self.interrogator.interrogate_fast(image)
+            clip_classic = self.interrogator.interrogate_classic(image)
             
             return full_prompt, clip_fast, clip_classic