ImportError: Using the `Trainer` with `PyTorch` requires `accelerate>=0.20.1`: Please run `pip install transformers[torch]` or `pip install accelerate -U'

merong·2023년 8월 23일
0

⚠ERROR! 모음zip

목록 보기
3/5

👾사건의 전말

허깅페이스에서 GPT 모델을 가져다가 fine-tuning을 시도했었는데,

from transformers import Trainer, TrainingArguments

training_args = TrainingArguments(
				...
                )
trainer - Trainer(
				...
                )

위의 코드에서 에러가 발생했다.

ImportError: Using the Trainer with PyTorch requires accelerate>=0.20.1: Please run pip install transformers[torch] or `pip install accelerate -U'

✅해결방법

에러에 나온대로, pip install transformers[torch] or pip install accelerate -U
를 시도하면 에러가 고쳐지지 않았다..

우선, 코드 첫 부분에

! pip install -U accelerate
! pip install -U transformers

위의 두 줄을 넣어주고, 런타임을 한번 삭제하고 재실행해야 한다.
그리고 추가한 코드부터 차례대로 코드를 실행시켜주면 된다....🥹

별거 없다... ㅋㅋㅋㅋㅋ

profile
매일매일이 새로운 시작점

0개의 댓글