Error AdamW RAGPretrainedModel

[helderalvesoliveira]

ImportError Traceback (most recent call last)

in <cell line: 0>()
---> 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 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)

[MoOo2mini]

#269 (comment)

[SuhaasBhat]

Newer huggingface versions (after approximately 4.49.0) have removed transformers.AdamW, and suggest importing AdamW from torch. Can we please update RAGatouille to be compatible with this?

[KundanSaha-png]

ImportError: cannot import name 'AdamW' from 'transformers'

[kalinduaet]

ImportError: cannot import name 'AdamW' from 'transformers'

use transformers==4.42.4