Update app.py

app.py

@@ -5,14 +5,6 @@ mod = types.ModuleType('flash_attn')
 mod.__spec__ = spec
 sys.modules['flash_attn'] = mod
 
-import transformers.utils.import_utils as _import_utils
-from transformers.utils import is_flash_attn_2_available
-_import_utils._is_package_available   = lambda pkg: False
-_import_utils.is_flash_attn_2_available = lambda: False
-
-transformers.utils.is_flash_attn_2_available              = getattr(transformers.utils, "is_flash_attn_2_available", lambda: False)
-transformers.utils.is_flash_attn_greater_or_equal_2_10    = lambda *args, **kwargs: False
-
 import huggingface_hub as _hf_hub
 _hf_hub.cached_download = _hf_hub.hf_hub_download
 
@@ -27,6 +19,14 @@ try:
 except ImportError:
     from diffusers import EulerDiscreteScheduler as FlowMatchEulerDiscreteScheduler
 
+import transformers.utils.import_utils as _import_utils
+from transformers.utils import is_flash_attn_2_available
+_import_utils._is_package_available   = lambda pkg: False
+_import_utils.is_flash_attn_2_available = lambda: False
+
+transformers.utils.is_flash_attn_2_available              = getattr(transformers.utils, "is_flash_attn_2_available", lambda: False)
+transformers.utils.is_flash_attn_greater_or_equal_2_10    = lambda *args, **kwargs: False
+
 REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
 
 # Florence-2 로드