Which path to use for "from model.P5 import P5"

[mgulcin]

I'm using the release branch (assuming it is stable) and trying to run an example test. I have only a single gpu, hence changed the test command with: ...--distributed 0 --gpu 0...

While running the command I receive the following error:

Traceback (most recent call last):
  File "/content/drive/MyDrive/OpenP5-release-1.0/test_command/../src/main.py", line 17, in <module>
    from model.P5 import P5
ModuleNotFoundError: No module named 'model.P5'

In model directory, there are "P5_T5.py" or "P5_LLAMA.py" files. Which one should be used? Or what is the right way for running the test commands?