Τον Μάρτιο του 1945, 6η Στρατιά Πάντσερ SS και η LSSAH πρωτοστάτησαν στην Επιχείρηση Ξύπνημα της Άνοιξης, μια επίθεση στην Ουγγαρία κοντά στη λίμνη Μπάλατον με στόχο την εξασφάλιση των τελευταίων πετρελαϊκών αποθεμάτων που εξακολουθούσαν να είναι διαθέσιμα για την Γερμανία. Παρά τις πρώτες νίκες, το σχέδιο ήταν υπερβολικά φιλόδοξο και απέτυχε. Μετά την αποτυχία αυτή, η 6η Στρατιά Πάντσερ SS και η LSSAH υποχώρησαν στην περιοχή της Βιέννης. Ως σημάδι ντροπής, οι μονάδες Waffen-SS που συμμετείχαν στη μάχη διατάχθηκαν από τον Χίτλερ για να αφαιρέσουν τα επιράμματα πού έφεραν το όνομα του Αδόλφου Χίτλερ. Ο Ντίτριχ δεν ανάφερε τη διαταγή στα στρατεύματά του και λίγο αργότερα, τα στρατεύματα του Ντίτριχ αναγκάστηκαν να υποχωρήσουν από τη Βιέννη από τις δυνάμεις του Σοβιετικού Στρατού. Ο Ντίτριχ, συνοδευόμενος από τη σύζυγό του, παραδόθηκε στις 9 Μαΐου 1945 στην αμερικανική 36η Μεραρχία Πεζικού στην Αυστρία.