시작하기무료로 시작하기

사전 학습된 PyTorch 모델로 언어 번역하기

PyBooks 팀은 한 언어에서 다른 언어로 번역하는 AI 프로젝트를 진행하고 있어요. 이 작업에 사전 학습된 모델을 활용하면 학습 시간과 자원을 크게 절약할 수 있습니다. 이번 연습 문제의 목표는 HuggingFace Transformers 라이브러리의 T5(Text-To-Text Transfer Transformer) 모델을 설정하고, 이를 사용해 영어 문장을 프랑스어로 번역하는 것입니다.

T5Tokenizer, T5ForConditionalGeneration는 미리 불러와 두었습니다.

이 연습은 강의의 일부입니다

PyTorch로 배우는 텍스트 딥러닝

강의 보기

연습 안내

  • 사전 학습된 "t5-small" 모델에서 tokenizermodel을 초기화하세요.
  • 토크나이저를 사용해 입력 프롬프트를 인코딩하고, PyTorch 텐서를 반환하도록 설정하세요.
  • model을 사용해 입력 프롬프트를 번역하고, 번역된 출력을 생성하세요.

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Initalize tokenizer and model
tokenizer = ____.from_pretrained("t5-small")
model = ____.from_pretrained("t5-small")

input_prompt = "translate English to French: 'Hello, how are you?'"

# Encode the input prompt using the tokenizer
input_ids = ____.____(input_prompt, return_tensors="____")

# Generate the translated ouput
output = model.____(input_ids, max_length=50)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print("Generated text:",generated_text)
코드 편집 및 실행