update roadmap

Author: harveyp123

ROADMAP.md

@@ -18,5 +18,5 @@
 - [ ] [lightllm](https://github.com/ModelTC/lightllm)
 
 ## Research
-- [ ] Optimize the tree-based attention to reduce additional computation
+- [x] Optimize the tree-based attention to reduce additional computation
 - [ ] Improve the acceptance scheme to generate more diverse sequences

pyproject.toml

@@ -24,4 +24,4 @@ train = ["bitsandbytes", "wandb", "scipy"]
 "Blog" = "https://sites.google.com/view/medusa-llm"
 
 [tool.setuptools.packages.find]
-exclude = ["assets*", "notebooks*", "scripts*"]
+exclude = ["assets*", "notebooks*", "scripts*", "llm_judge"]