I confirm you can use any capacity Li-Ion battery that is the correct voltage. (I am an electronic technician.)
For some quads there are various capacity batteries available, but from experience they are not always the capacity claimed !
(You can check battery capacity when charging through a...