Fix: register LLM as nn.Module submodule so optimizer finds LoRA params cdc806e Faaz commited on Apr 15