beyoru commited on
Commit
b950192
·
verified ·
1 Parent(s): 619b16f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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"👉 Router chọn expert: {chosen_expert}")
75
 
76
  # Generate với expert được chọn
77
- output_ids = model.generate(chat_prompt, max_new_tokens=256)
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
  ```