Update modeling_llava_qwen2.py

modeling_llava_qwen2.py

@@ -19,9 +19,10 @@ from transformers.image_utils import (ChannelDimension, PILImageResampling, to_n
 from transformers.modeling_outputs import BaseModelOutput, BaseModelOutputWithPooling
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import ModelOutput
+import subprocess
 
 torch.set_default_device('cuda')
-
+subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 class SigLipImageProcessor:
     def __init__(self,