transformers

ba95ec66 - [distributed] Reinstall the context parallel attention wrapper when its key was registered again