ImportError: cannot import name 'LoggingArgs' from partially initialized module 'nanotron.config.config error (circular imports)

[Akhilvallala2023]

Description:
I encountered an ImportError related to a circular import in the nanotron library when trying to run my training script in Google Colab. The error message is as follows:
text
"""""""ImportError: cannot import name 'LoggingArgs' from partially initialized module 'nanotron.config.config'"""""""

Steps to Reproduce:
Set up a Google Colab environment with GPU.
Install the required libraries using:
bash
!pip install torch transformers datasets matplotlib nanotron

Run the training script train_custom_smollm.py with the configuration file custom_smollm_config.yaml.
Expected Behavior:
The script should run without any import errors, allowing for successful model training.
Environment:
Python version: 3.10

Google Colab environment
Additional Context:
I have attached screenshots of my setup and the error message.
I attempted reinstalling the library and checking for updates, but the issue persists.
Please let me know if there's a workaround or if further information is needed. You can fill in specific details like the version of nanotron if you have it.

I am getting import error, i don;t know, where I am getting wrong, I did run the code as per the example.