`depth` variable isn't used in `run_lats` function

[knotgrass]

I try to run your demo in hugginface space. I realize that depth variable isn't used in run_lats function

Otherwise, the depth variable is used as tree_depth
May you fix it?
Thank you.