On January 4, 1945, Guards Colonel E.Ya. Birstein was given command of the division, which he would hold for the duration. At about the same time the 334th made its final transfer, to 2nd Guards Army in 3rd Belorussian Front, still in 60th Rifle Corps. Under these commands it took part in the Vistula-Oder Offensive. When the offensive was launched, 60th Rifle Corps was assigned to defend along a broad front to the north of Goldap while the remainder of 2nd Guards Army followed 28th Army's breakthrough of the German lines; the 334th and its Corps followed in second echelon. In the second half of March, in the lead-up to the assault on Königsberg, 60th Corps advanced along a route from Preussisch Eylau to Domnau, Tapiau, and Granz. On April 5 the division was awarded the Order of Suvorov, 2nd Class, in recognition of its role in the capture of Bartenstein and Landsberg. During the assault on Königsberg on April 7, the division assisted in forcing the Tierenberger River and reached the area north of Norgau. Once the city fell, the 334th took part in the clearing of the Samland Peninsula beginning on April 13.