TypeError: 'NoneType' object cannot be interpreted as an integer

I'm having some issues running the following notebook locally:

https://github.com/AI4Finance-Foundation/FinGPT/blob/master/FinGPT_Training_LoRA_with_ChatGLM2_6B_for_Beginners.ipynb

1. It looks like there are minor differences between the file in the repository and the one on Google Colab.
2. The notebook an Google Colab seems to be failing despite installing and importing bitsandbytes and accelerate.
3. I tried to run the notebook locally, but run out of memory, originally assuming that the pretrained model called in the example is not multii-gpu. (For this reason I used Llama, which I downloaded locally manually.)
4. I'm now running into issues in the training stage. I'm unsure why the labels include None values? Maybe this is being caused by a change I made, but I do not have a working baseline to inspect (since the google colab file is not working).

Attached is my copy of the notebook. Wondering if someone can provide some helpful pointers. Been staring at this for a while.
[FinGPT_Training_LoRA_with_ChatGLM2_6B_for_Beginners.zip](https://github.com/AI4Finance-Foundation/FinGPT/files/13537649/FinGPT_Training_LoRA_with_ChatGLM2_6B_for_Beginners.zip)