arg names

This commit is contained in:
Raphael Walker 2024-12-05 16:05:15 +01:00
parent e716b529f7
commit 57cc24dca8

View File

@ -72,7 +72,7 @@ class _ReduxAttnWrapper:
x, c = self.previous(img=args["img"], txt=args["txt"],vec=args["vec"], pe=args["pe"], attn_mask=mask)
return {"img": x, "txt": c}
case SingleStreamBlock():
x = self.previous(img=args["img"], vec=args["vec"], pe=args["pe"], attn_mask=mask)
x = self.previous(args["img"], vec=args["vec"], pe=args["pe"], attn_mask=mask)
return {"img": x}
case _ReduxAttnWrapper():
# pass along the mask, and tell the next redux what its part of the mask is