Оба варианта написания являются правильными. "Щенки" используется в единственном числе и множественном числе в именительном и винительном падеже, а "щенята" - в родительном, дательном, творительном и предложном падежах единственного числа.