Error AdamW RAGPretrainedModel

ImportError Traceback (most recent call last)

[](https://localhost:8080/#) in ()
---> 1 from ragatouille import RAGPretrainedModel
2
3 RAG = RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")

6 frames

[/usr/local/lib/python3.11/dist-packages/colbert/training/training.py](https://localhost:8080/#) in
5 import numpy as np
6
---> 7 from transformers import AdamW, get_linear_schedule_with_warmup
8 from colbert.infra import ColBERTConfig
9 from colbert.training.rerank_batcher import RerankBatcher

ImportError: cannot import name 'AdamW' from 'transformers' (/usr/local/lib/python3.11/dist-packages/transformers/__init__.py)

Start by reproducing the import with `from ragatouille import RAGPretrainedModel` and `RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")`. Then inspect `colbert/training/training.py`, especially its `from transformers import AdamW` import, and verify that the model import completes without the reported ImportError.

Written by the indexing model from the issue text.