Some questions that arise when studying your code
Thank you for your excellent work! I learned a lot from it!
But I have a question when studying your code. Like your work with Medusa, there is a variable called tree_choices, mc_sim_7b_63, as well as tree_indices, retrieve_indices. Can you share what these variables mean?
I have run Medusa code and trained the draft model. I found that it also needs to run the script to generate corresponding choices for the trained model. It seems to be the path of the tree? I don't quite understand the meaning of this. This variable needs to be imported into choices.py
In EAGLE, mc_sim_7b_63 represents the structure of the tree; see #6 for details. In EAGLE-2, the tree structure is dynamically determined based on the context, so this variable has been deprecated.