unlimiformer: ImportError: cannot import name 'Unlimiformer' from 'unlimiformer'

Hi, I’ve been trying to run inference using the model checkpoints on HF by following the code in inference-example.py.

I’m using Google Colab and I’m cloning the whole repository. However, when I try to run the code, I get the following error:

---------------------------------------------------------------------------

ImportError                               Traceback (most recent call last)

[<ipython-input-9-ea5aa8573cba>](https://localhost:8080/#) in <cell line: 1>()
----> 1 from unlimiformer import Unlimiformer
      2 from random_training_unlimiformer import RandomTrainingUnlimiformer
      3 from usage import UnlimiformerArguments, training_addin
      4 
      5 from transformers import BartForConditionalGeneration, AutoTokenizer

ImportError: cannot import name 'Unlimiformer' from 'unlimiformer' (unknown location)


---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

I’ve tried multiple things but nothing seems to work:

• Restarting runtime
• Deleting runtime & creating a new one
• Adding the directory to PATH
• Changing to directory using %cd /content/unlimiformer/src
• Moving the files to /content/

I’m not sure what I’m missing here. It seems to be such a minor issue but I’m out here pulling my hair out trying to figure why it isn’t working. If someone can help me out I’d really appreciate it. Thanks!