Update util/utils.py

util/utils.py

@@ -59,12 +59,25 @@ def get_caption_model_processor(model_name, model_name_or_path="Salesforce/blip2
             model_name_or_path, device_map=None, torch_dtype=torch.float16
         ).to(device)
     elif model_name == "florence2":
+        # Fixed: Use AutoProcessor and AutoModelForCausalLM to avoid deprecation warning
         from transformers import AutoProcessor, AutoModelForCausalLM 
+        
+        # Keep both processors - base for processor, ft for model
         processor = AutoProcessor.from_pretrained("microsoft/Florence-2-base", trust_remote_code=True)
+        
         if device == 'cpu':
-            model = AutoModelForCausalLM.from_pretrained(model_name_or_path, torch_dtype=torch.float32, trust_remote_code=True)
+            # Use AutoModelForCausalLM instead of direct Florence2 class
+            model = AutoModelForCausalLM.from_pretrained(
+                model_name_or_path, 
+                torch_dtype=torch.float32, 
+                trust_remote_code=True
+            )
         else:
-            model = AutoModelForCausalLM.from_pretrained(model_name_or_path, torch_dtype=torch.float16, trust_remote_code=True).to(device)
+            model = AutoModelForCausalLM.from_pretrained(
+                model_name_or_path, 
+                torch_dtype=torch.float16, 
+                trust_remote_code=True
+            ).to(device)
     return {'model': model.to(device), 'processor': processor}