diff --git a/nodes.py b/nodes.py
index 5bb1834..9f797ff 100644
--- a/nodes.py
+++ b/nodes.py
@@ -78,6 +78,7 @@ class DownloadAndLoadPyramidFlowModel:
             text_encoder_dtype,
             vae_dtype,
             model_variant=variant,
+            use_flash_attn=use_flash_attn,
         )
        
         # #fp8