gitadmin/comfyui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: sed patch xflux DoubleStreamBlock.forward for attn_mask compat

Browse Source
This commit is contained in:
gitadmin
2026-06-22 18:16:17 +00:00
parent 81a6e6f150
commit 65cd147189
1 changed files with 3 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+3 -11
View File
@@ -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"]