From 65cd147189831f55894d6bcffc04d2437d1dd0b0 Mon Sep 17 00:00:00 2001 From: gitadmin Date: Mon, 22 Jun 2026 18:16:17 +0000 Subject: [PATCH] fix: sed patch xflux DoubleStreamBlock.forward for attn_mask compat --- Dockerfile | 14 +++----------- 1 file changed, 3 insertions(+), 11 deletions(-) diff --git a/Dockerfile b/Dockerfile index d51078a..ec6c055 100644 --- a/Dockerfile +++ b/Dockerfile @@ -21,19 +21,11 @@ RUN cd /app/custom_nodes \ && git clone --depth 1 https://github.com/XLabs-AI/x-flux-comfyui \ && pip install --no-cache-dir -r x-flux-comfyui/requirements.txt -# Fix: xflux DoubleStreamBlock.forward must accept attn_mask kwarg passed by ComfyUI -RUN python3 -c " -path = '/app/custom_nodes/x-flux-comfyui/xflux/src/flux/modules/layers.py' -with open(path) as f: c = f.read() -old = ' def forward(self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor) -> tuple[Tensor, Tensor]:\n return self.processor(self, img, txt, vec, pe)' -new = ' def forward(self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor, attn_mask=None, transformer_options={}, **kwargs):\n return self.processor(self, img, txt, vec, pe)' -assert old in c, 'Pattern not found!' -with open(path, 'w') as f: f.write(c.replace(old, new)) -print('Patch applied') -" +RUN sed -i 's/def forward(self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor) -> tuple\[Tensor, Tensor\]:/def forward(self, img: Tensor, txt: Tensor, vec: Tensor, pe: Tensor, attn_mask=None, transformer_options={}, **kwargs):/' /app/custom_nodes/x-flux-comfyui/xflux/src/flux/modules/layers.py \ + && grep "attn_mask" /app/custom_nodes/x-flux-comfyui/xflux/src/flux/modules/layers.py RUN useradd -u 1000 -m comfyui && chown -R 1000:1000 /app USER 1000 EXPOSE 8188 -CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"] +CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"] \ No newline at end of file