Update src/florence/model.py

src/florence/model.py

@@ -24,7 +24,7 @@ from src.exception import CustomExceptionHandling
 model_id = "microsoft/Florence-2-large-ft"
 try:
     model = (
-        AutoModelForCausalLM.from_pretrained(model_id, attn_implementation="sdpa", torch_dtype=torch.float16, trust_remote_code=True)
+        AutoModelForCausalLM.from_pretrained(model_id, attn_implementation="eager", torch_dtype=torch.float16, trust_remote_code=True)
         .to("cuda")
         .eval()
     )