Update main_process/moe_router.py

main_process/moe_router.py

@@ -60,6 +60,7 @@ class NState(dict):
 
 # ---------------- LLM utilizado para el free_narration ----------------
 class SalamandraClient:
+    '''
     def __init__(self, model_id="BSC-LT/salamandra-7b-instruct"):
         self.tokenizer = AutoTokenizer.from_pretrained(model_id)
         self.model = AutoModelForCausalLM.from_pretrained(
@@ -67,6 +68,7 @@ class SalamandraClient:
             device_map="auto",
             torch_dtype=torch.bfloat16
         )
+    '''
 
     def chat(self, prompt) -> str:
         return get_from_prompt(prompt)