llm

Sleeping

Chris4K commited on Jan 27

Commit

f9528ef

verified ·

1 Parent(s): 311bab5

Update services/llama_generator.py

Files changed (1) hide show

services/llama_generator.py CHANGED Viewed

@@ -85,7 +85,7 @@ class LlamaGenerator(BaseGenerator):
         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()

         if not self.model:
             raise ValueError(f"Failed to load model: {llama_model_name}")
+        self.prm_model = self.model_manager.models.get("prm")
         self.prompt_builder = LlamaPromptTemplate()