From e05cd55498e7246786b6c2c94e8340c0795cf48c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jukka=20Sepp=C3=A4nen?= <40791699+kijai@users.noreply.github.com> Date: Thu, 10 Oct 2024 22:12:35 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 1 + 1 file changed, 1 insertion(+) 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