Språköversättning med en förtränad PyTorch-modell
Ditt team på PyBooks arbetar med ett AI-projekt som innebär översättning mellan olika språk. De vill använda förtränade modeller för uppgiften, vilket kan spara både tid och resurser. I den här övningen ska du konfigurera en översättningsmodell från HuggingFaces Transformers-bibliotek – närmare bestämt T5-modellen (Text-To-Text Transfer Transformer) – och använda den för att översätta en engelsk fras till franska.
T5Tokenizer och T5ForConditionalGeneration har redan laddats in åt dig.
Den här övningen är en del av kursen
Djupinlärning för text med PyTorch
Övningsinstruktioner
- Initiera
tokenizerochmodelfrån den förtränade modellen"t5-small". - Koda inmatningsprompten med tokenizern och se till att returnera PyTorch-tensorer.
- Översätt inmatningsprompten med
modeloch generera den översatta utdatan.
Interaktiv övning med praktiskt arbete
Testa den här övningen genom att slutföra den här exempelkoden.
# 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)