ERROR in train_model.ipynb on Colab

[PKUTHM]

Description:

I encountered an error while trying to run the train_model.ipynb notebook provided in the official RelBench repository on Google Colab. Specifically, I am unable to import SentenceTransformer from the sentence_transformers library.

Error Message:

from sentence_transformers import SentenceTransformer

RuntimeError: Failed to import transformers.integrations.integration_utils because of the following error (look up to see its traceback):
Failed to import transformers.modeling_utils because of the following error (look up to see its traceback):
partially initialized module 'torchvision' has no attribute 'extension' (most likely due to a circular import)

Attachment

train_model.zip
I have attached the Colab notebook (train_model.ipynb) with ERROR message downloaded from Colab as a reference.

[rishabh-ranjan]

@XinweiHe @zechengz can you please look into this?