Update src/florence/model.py

src/florence/model.py

@@ -14,11 +14,11 @@ from src.exception import CustomExceptionHandling
 
 
 # Install the required dependencies
-# subprocess.run(
-#     "pip install flash-attn --no-build-isolation",
-#     env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
-#     shell=True,
-# )
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
 
 # Load model and processor from Hugging Face
 model_id = "microsoft/Florence-2-large-ft"