Sprachübersetzung mit einem vortrainierten PyTorch-Modell
Dein Team bei PyBooks arbeitet an einem KI-Projekt, das die Übersetzung von einer Sprache in eine andere umfasst. Für diese Aufgabe sollen vortrainierte Modelle genutzt werden, was viel Trainingszeit und Ressourcen spart. In dieser Übung richtest du ein Übersetzungsmodell aus der Transformers-Bibliothek von HuggingFace ein, konkret das T5 (Text-To-Text Transfer Transformer)-Modell, und verwendest es, um einen englischen Satz ins Französische zu übersetzen.
T5Tokenizer, T5ForConditionalGeneration wurden für dich geladen.
Diese Übung ist Teil des Kurses
Deep Learning für Text mit PyTorch
Anleitung zur Übung
- Initialisiere
tokenizerundmodelaus dem vortrainierten Modell"t5-small". - Kodiere die Eingabeaufforderung mit dem Tokenizer und gib dabei sicher PyTorch-Tensoren zurück.
- Übersetze die Eingabeaufforderung mit
modelund generiere die übersetzte Ausgabe.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# 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)