update llama.py

vllm/model_executor/models/llama.py

@@ -455,6 +455,7 @@ class LlamaForCausalLM(nn.Module, SupportsLoRA):
             self.quant_config=quant_config
               
         self.use_llama_nn = os.environ.get('LLAMA_NN') == '1'
+        self.use_lm_nn = os.environ.get('LM_NN') == '1'
         self.use_gemm_pad = os.environ.get('GEMM_PAD') == '1'
         self.use_fa_pad = os.environ.get('FA_PAD') == '1'
         self.use_awq_pad = os.environ.get('AWQ_PAD') == '1'