[BUG] Nanotron runner imports non-existant

[jquesnelle]

Describe the bug

Attempting to run a nanotron eval (lighteval nantron) will result in an error since hierarchical_logger does not exist. See https://github.com/huggingface/lighteval/blob/main/src/lighteval/main_nanotron.py#L55

To Reproduce

Run

lighteval nanotron --checkpoint-config-path some-nanotron-path --lighteval-config-path some-lighteval-path

Expected behavior

Nanotron eval works

Version info

Reproduced on fac17bb

[jquesnelle]

Follow up -- if you remove hierarchical_logger, https://github.com/huggingface/lighteval/blob/main/src/lighteval/pipeline.py#L73 errors. Looks like it was renamed to lighteval.models.nanotron.nanotron_model

[peteboyd]

Have this issue too. Can confirm that the fixes provided by @jquesnelle work. 1) remove the htrack_block import and code block in main_nanotron.py and 2) rename the import in pipeline.py.

[NathanHB]

great ! thanks for the fix, can you open a PR so that we can review and merge ? :)