Update app.py

app.py

@@ -43,7 +43,7 @@ class PromptEnhancerV2:
         self.model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
             models_root_path,
             torch_dtype=dtype,
-            attn_implementation="flash_attention_2",
+            # attn_implementation="flash_attention_2",
             device_map=device_map,
         )
         self.processor = AutoProcessor.from_pretrained(models_root_path)