SafeAILab / EAGLE

Official Implementation of EAGLE
https://arxiv.org/pdf/2406.16858
Apache License 2.0
622 stars 59 forks source link

The initiization of tree_buffers #68

Closed kafkayu closed 1 month ago

kafkayu commented 1 month ago

Thanks for your excellent work. I want to change the structure of choices tree,but i could not find the defination of some configuration paramters of tree_buffers ,such as retrieve_indices. image Function "generate_tree_buffers" does not initialize the parameters mentioned above.If I want to create a chain structure of choices tree, specifically, [[0],[0,0],[0,0,0],[0,0,0,0],[0,0,0,0,0]] , how should I do?