Update main_process/moe_router.py
Browse files
main_process/moe_router.py
CHANGED
|
@@ -59,7 +59,6 @@ class DataHub:
|
|
| 59 |
class NState(dict):
|
| 60 |
pass
|
| 61 |
|
| 62 |
-
# ---------------- LLM utilizado para el free_narration ----------------
|
| 63 |
class SalamandraClient:
|
| 64 |
def __init__(self, model_id="BSC-LT/salamandra-7b-instruct-tools"):
|
| 65 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=HF_TOKEN)
|
|
@@ -86,7 +85,6 @@ class SalamandraClient:
|
|
| 86 |
|
| 87 |
return self.tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 88 |
|
| 89 |
-
# Esto aqu铆 s贸lo se utiliza para la valoraci贸n:
|
| 90 |
class GPT5Client:
|
| 91 |
def __init__(self, api_key: str):
|
| 92 |
key = api_key
|
|
|
|
| 59 |
class NState(dict):
|
| 60 |
pass
|
| 61 |
|
|
|
|
| 62 |
class SalamandraClient:
|
| 63 |
def __init__(self, model_id="BSC-LT/salamandra-7b-instruct-tools"):
|
| 64 |
self.tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=HF_TOKEN)
|
|
|
|
| 85 |
|
| 86 |
return self.tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 87 |
|
|
|
|
| 88 |
class GPT5Client:
|
| 89 |
def __init__(self, api_key: str):
|
| 90 |
key = api_key
|