uses_xformers#

core.nn.uses_xformers(impl)[source]#

What to pass as use_xformers when constructing the attention modules.

xformers is an optional extra, and torch_brain only notices it is missing on the first forward; check here so a mis-provisioned env fails at construction instead of after the data pipeline has spun up.