Update README.md
Browse files
README.md
CHANGED
|
@@ -71,10 +71,10 @@ print(chat_prompt)
|
|
| 71 |
|
| 72 |
# Check router decision
|
| 73 |
chosen_expert = model.route(messages[-1]["content"])
|
| 74 |
-
print(f"
|
| 75 |
|
| 76 |
# Generate với expert được chọn
|
| 77 |
-
output_ids = model.generate(chat_prompt, max_new_tokens=
|
| 78 |
print(tokenizer.decode(output_ids[0], skip_special_tokens=False))
|
| 79 |
|
| 80 |
```
|
|
|
|
| 71 |
|
| 72 |
# Check router decision
|
| 73 |
chosen_expert = model.route(messages[-1]["content"])
|
| 74 |
+
print(f"Router chosen: {chosen_expert}")
|
| 75 |
|
| 76 |
# Generate với expert được chọn
|
| 77 |
+
output_ids = model.generate(chat_prompt, max_new_tokens=1024)
|
| 78 |
print(tokenizer.decode(output_ids[0], skip_special_tokens=False))
|
| 79 |
|
| 80 |
```
|