addede a Production Branch

image_indexer.py

@@ -322,19 +322,15 @@ class ImageIndexer:
             # Load model and processor with proper device handling
             self.processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
             
-            # Load model directly to the target device to avoid meta tensor issues
-            if self.device == "cuda":
-                self.model = CLIPModel.from_pretrained(
-                    "openai/clip-vit-base-patch16",
-                    torch_dtype=torch.float16,
-                    device_map="auto"
-                )
-            else:
-                # For CPU, use device_map to avoid meta tensor issues
-                self.model = CLIPModel.from_pretrained(
-                    "openai/clip-vit-base-patch16",
-                    device_map="cpu"
-                )
+            # Load model with explicit device placement to avoid meta tensor issues
+            self.model = CLIPModel.from_pretrained(
+                "openai/clip-vit-base-patch16",
+                torch_dtype=torch.float32,  # Use float32 for CPU compatibility
+                device_map=None  # Don't use device_map to avoid meta tensor issues
+            )
+            
+            # Move model to device explicitly
+            self.model = self.model.to(self.device)
             
             self.model_initialized.set()
             print("Model initialization complete")

image_search.py

@@ -16,19 +16,17 @@ class ImageSearch:
         # Load model and processor with proper device handling
         self.processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")
         
-        # Load model directly to the target device to avoid meta tensor issues
-        if self.device == "cuda":
-            self.model = CLIPModel.from_pretrained(
-                "openai/clip-vit-base-patch16",
-                torch_dtype=torch.float16,
-                device_map="auto"
-            )
-        else:
-            # For CPU, use device_map to avoid meta tensor issues
-            self.model = CLIPModel.from_pretrained(
-                "openai/clip-vit-base-patch16",
-                device_map="cpu"
-            )
+        # Load model with explicit device placement to avoid meta tensor issues
+        self.model = CLIPModel.from_pretrained(
+            "openai/clip-vit-base-patch16",
+            torch_dtype=torch.float32,  # Use float32 for CPU compatibility
+            device_map=None  # Don't use device_map to avoid meta tensor issues
+        )
+        
+        # Move model to device explicitly
+        self.model = self.model.to(self.device)
+        
+        print("Model initialization complete")
         
         # Initialize Qdrant client, folder manager and image database
         self.qdrant = QdrantClientSingleton.get_instance()