Клонирование GitHub вызывает ошибку ModuleNotFoundError в Google Colab ⇐ Python

[Anonymous]

Я пытаюсь клонировать репозиторий, известный как progen3, на GitHub и создать его в своей среде Google Colab. Однако всякий раз, когда я это делаю, а затем пытаюсь получить доступ к элементам из репозитория, это вызывает ошибку ModuleNotFoundError.
Я начал с обычного клонирования:
!git clone https://github.com/Profluent-AI/progen3.git
%cd progen3
!bash setup.sh

а затем в новой ячейке использовал этот код:
import torch

from progen3.modeling import ProGen3ForCausalLM
from progen3.batch_preparer import ProGen3BatchPreparer
from progen3.scorer import ProGen3Scorer

model = ProGen3ForCausalLM.from_pretrained("Profluent-Bio/progen3-3b", torch_dtype=torch.bfloat16)
model = model.eval().to("cuda:0")
batch_preparer = ProGen3BatchPreparer()

# Direct Usage
sequence = "MALWMRLLPLLALLALWGPDPAAAFVNQHLCGSHLVEALYLVCGERGFFYTPKTRREAEDLQVGQVELGGGPGAGSLQPLALEGSLQKRGIVEQCCTSICSLYQLENYCN"

inputs = batch_preparer.get_batch_kwargs([sequence], device="cuda:0", reverse=False)
outputs = model(**inputs, return_dict=True)
print(outputs.logits)

# Usage with scorer (returns averaged log likelihood of forward and reverse direction)
# Would suggest using Scoring CLI below if scoring very large number of sequences
scorer = ProGen3Scorer(model=model)
scores = scorer.score_batch(sequences=[sequence])
print(scores["log_likelihood"][0])

Это приводит к следующей ошибке:
---------------------------------------------------------------------------
ModuleNotFoundError Traceback (most recent call last)
/tmp/ipython-input-4120543976.py in ()
1 import torch
2
----> 3 from progen3.modeling import ProGen3ForCausalLM
4 from progen3.batch_preparer import ProGen3BatchPreparer
5 from progen3.scorer import ProGen3Scorer

ModuleNotFoundError: No module named 'progen3.modeling'

---------------------------------------------------------------------------
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.
---------------------------------------------------------------------------

Это пример кода из README.md в репозитории progen3. Я не уверен, почему это не работает или что создает эту проблему. Возможно, это связано с небольшими ошибками при импорте. Конкретной ошибки не возникает, но она отмечена красным текстом:
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
google-cloud-bigquery 3.38.0 requires packaging>=24.2.0, but you have packaging 24.1 which is incompatible.
db-dtypes 1.4.4 requires packaging>=24.2.0, but you have packaging 24.1 which is incompatible.

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
torchaudio 2.9.0+cu126 requires torch==2.9.0, but you have torch 2.5.1 which is incompatible.
torchvision 0.24.0+cu126 requires torch==2.9.0, but you have torch 2.5.1 which is incompatible.


Подробнее здесь: https://stackoverflow.com/questions/798 ... ogle-colab