Revert "fix(CRITICAL): 极简防弹Dockerfile——每个RUN都 || true"

This reverts commit a8acfbbb8f.
This commit is contained in:
Claude
2026-08-11 18:08:53 +00:00
parent a8acfbbb8f
commit f580b14dc3
135 changed files with 10152 additions and 31633 deletions

View File

@@ -942,10 +942,18 @@ class XFormersImpl(AttentionImpl[XFormersMetadata]):
query = query.unsqueeze(0)
key = key.unsqueeze(0)
value = value.unsqueeze(0)
# BI-V100: ixformer varlen_fwd has incompatible signature with
# xops.fmha.flash.FwOp() (20-arg mismatch). Use pure-math fallback
# for ALL head sizes during prefill, not just head_size > 128.
out = self._run_sdpa_fallback(query, key, value, attn_metadata)
if self.head_size > 128:
out = self._run_sdpa_fallback(query, key, value, attn_metadata)
else:
out = xops.memory_efficient_attention_forward(
query,
key,
value,
attn_bias=attn_bias[0],
p=0.0,
scale=self.scale,
op=self.attn_op,
)
return out.view_as(original_query)
# Attention with alibi slopes.