Update README.md

README.md

@@ -80,10 +80,10 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True,
                                           use_fast=False,
                                          )
 # Load critic model
-model_name_base = "meta-llama/Llama-3.2-3B-Instruct"
+model_name_critic = "meta-llama/Llama-3.2-3B-Instruct"
 
 critic_model = AutoModelForCausalLM.from_pretrained(
-    model_name_base, 
+    model_name_critic, 
     torch_dtype=torch.bfloat16, 
     attn_implementation="flash_attention_2", 
     device_map="auto",