[PyTorch] 사내망에서 BERT 불러오는 방법

qw4735·2024년 5월 20일
0

PyTorch

목록 보기
8/8

사내망에서 bert 불러오기

  1. huggingface에서 모델 파일 전부 다운로드 -> 모델 안에 kykim\bert-kor-base 폴더 만들고 그 안에 다운로드 받은 파일 전부 저장
    https://huggingface.co/kykim/bert-kor-base/tree/main

  2. 모델 불러올때 해당 경로 지정 및 local_files_only=True 설정

    from  transformers import AutoModelForSequenceClassification
    
    checkpoint = "kykim/bert-kor-base"   # 상대경로
    model = AutoModelForSequenceClassification.from_pretrained(checkpoint, local_files_only=True, num_labels=4).to(device)

0개의 댓글