NameError: name 'get_key_to_not_convert' is not defined

Hi,

Iā€™m trying the colab notebook HF posted for running bloom on a colab: Google Colab

Getting ā€œNameError: name ā€˜get_key_to_not_convertā€™ is not definedā€, looks like this function is getting imported from bitsandbytes, but I canā€™t find it in the bitsandbytes repo.

Did anyone run this succesfully?