Update nodes.py

2024-10-10 22:12:35 +03:00
parent 754ef838d2
commit e05cd55498
1 changed files with 1 additions and 0 deletions
@@ -78,6 +78,7 @@ class DownloadAndLoadPyramidFlowModel:
            text_encoder_dtype,
            vae_dtype,
            model_variant=variant,
+            use_flash_attn=use_flash_attn,
        )
       
        # #fp8