By default, the application will automatically download media. But if you turn the feature off earlier, go to WhatsApp > More. Select Settings > Storage And Data > Media Auto-Download. Now from the three options; When Using Mobile Data, When Connected To Wi-Fi, or When Roaming, select the one.