사전 학습된 PyTorch 모델로 언어 번역하기
PyBooks 팀은 한 언어에서 다른 언어로 번역하는 AI 프로젝트를 진행하고 있어요. 이 작업에 사전 학습된 모델을 활용하면 학습 시간과 자원을 크게 절약할 수 있습니다. 이번 연습 문제의 목표는 HuggingFace Transformers 라이브러리의 T5(Text-To-Text Transfer Transformer) 모델을 설정하고, 이를 사용해 영어 문장을 프랑스어로 번역하는 것입니다.
T5Tokenizer, T5ForConditionalGeneration는 미리 불러와 두었습니다.
이 연습은 강의의 일부입니다
PyTorch로 배우는 텍스트 딥러닝
연습 안내
- 사전 학습된
"t5-small"모델에서tokenizer와model을 초기화하세요. - 토크나이저를 사용해 입력 프롬프트를 인코딩하고, 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)