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

fix: use fixes.py script for x-flux-comfyui patches (attn_mask + sdpa dtype)

Browse Source
This commit is contained in:
gitadmin
2026-06-22 18:43:04 +00:00
parent f5c7f97f69
commit 4345aeaf8a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Dockerfile
+2 -2
View File
@@ -21,8 +21,8 @@ RUN cd /app/custom_nodes \
&& git clone --depth 1 https://github.com/XLabs-AI/x-flux-comfyui \ && git clone --depth 1 https://github.com/XLabs-AI/x-flux-comfyui \
&& pip install --no-cache-dir -r x-flux-comfyui/requirements.txt && pip install --no-cache-dir -r x-flux-comfyui/requirements.txt
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 \ COPY fixes.py /tmp/fixes.py
&& grep "attn_mask" /app/custom_nodes/x-flux-comfyui/xflux/src/flux/modules/layers.py RUN python3 /tmp/fixes.py
RUN useradd -u 1000 -m comfyui && chown -R 1000:1000 /app RUN useradd -u 1000 -m comfyui && chown -R 1000:1000 /app