fix: return proper attention for llama4 lora kernel and fsdp2 llama4 example fix#2943

winglian merged 2 commits into

mainaxolotl-ai-cloud/axolotl:mainfrom

fix/minor-fixesaxolotl-ai-cloud/axolotl:fix/minor-fixesCopy head branch name to clipboard