Update vision.py
#9
by
Keerthi23
- opened
When flash attention IS available: The code imports the required modules as before
When flash attention is NOT available: The code now raises a clear ImportError with installation instructions
Benefits:
Provides immediate feedback when flash attention is missing
Gives clear installation instructions
Prevents silent failures or confusing errors later in the code