Update services/llama_generator.py

services/llama_generator.py

@@ -84,7 +84,10 @@ class LlamaGenerator(BaseGenerator):
         self.model = self.model_manager.models.get("llama")
         if not self.model:
             raise ValueError(f"Failed to load model: {llama_model_name}")
-            
+
+        self.model_prm = self.model_manager.models.get("prm")
+
+        
         self.prompt_builder = LlamaPromptTemplate()
         self._init_strategies()