Update test_ai_integration_http.py

test_ai_integration_http.py

@@ -120,23 +120,25 @@ def test_ai_integration_http():
                 )
                 status['processor_loaded'] = True
                 
-                # Get config and model class from the processor
-                config = processor.config
-                
-                # Load the model as a general vision model
+                # Load model directly - let it handle config internally
                 model = AutoModel.from_pretrained(
                     model_name,
                     trust_remote_code=True,
                     torch_dtype=torch.float32,  # Use float32 for better compatibility
-                    device_map=None  # Don't auto-map devices
+                    device_map=None,  # Don't auto-map devices
+                    ignore_mismatched_sizes=True  # Handle any size mismatches
                 )
                 status['model_loaded'] = True
                 
-                # Log model architecture
+                logger.info(f"Processor type: {type(processor).__name__}")
+                logger.info(f"Model type: {type(model).__name__}")
+                
+                # Log model architecture and details
                 model_size = get_model_size(model)
                 logger.info(f"Model loaded: {model_size/1e9:.2f} GB in parameters")
                 logger.info(f"Model architecture: {model.__class__.__name__}")
-                logger.info(f"Model config type: {type(config).__name__}")
+                if hasattr(model, 'config'):
+                    logger.info(f"Model config type: {type(model.config).__name__}")
             except Exception as e:
                 logger.error(f"Model loading failed: {str(e)}")
                 raise