Test model for LLaMA architecture