A.- La société X._ Sàrl a été constituée le 10 décembre 1998. A._, associé gérant, en est l'unique employé, en qualité de directeur. Par convention du 28 janvier 1999, la Caisse nationale suisse d'assurance en cas d'accidents (CNA) et la société ont fixé le gain assuré selon la LAA de A._, dès le 1er février 1999, à 5000 fr. par mois, à titre de salaire minimum correspondant aux usages professionnels et locaux. Au mois de juin 1999, la société a demandé à la CNA une réduction du gain assuré. Elle alléguait une situation financière difficile, l'obligeant à réduire sa masse salariale à 12 000 fr. par an dès le 1er juillet 1999. Ensuite du refus signifié, par lettre du 24 juin 1999, à la société, cette dernière a encore indiqué que A._ avait été licencié puis réengagé dès le 1er juillet 1999 avec un taux d'activité de 25 % pour 10 heures de travail hebdomadaires. Le 8 février 2000, la société a déclaré un revenu déterminant pour le calcul des primes définitives de l'années 1999 de 11 000 fr. pour la période du 1er février au 31 décembre 1999. Par décision du 15 mai 2000, confirmée sur opposition le 4 août 2000, la CNA a fixé le montant des cotisations pour l'année 1999 à 1257 fr., sur la base d'un salaire mensuel de 5000 fr. du 1er février au 30 juin et de 1250 fr. dès le 1er juillet 1999. B.- Par jugement du 2 avril 2001, le Tribunal des assurances du canton de Vaud a rejeté le recours formé par X._ Sàrl contre la décision sur opposition du 24 août 2000. C.- X._ Sàrl interjette recours de droit administratif contre ce jugement. Elle conclut, sous suite de frais et dépens, à ce que les primes de l'assurance obligatoire contre les accidents pour l'année 1999 soient fixées sur la base du salaire effectif de 11 000 fr. La CNA a conclu au rejet du recours, cependant que l'Office fédéral des assurances sociales a renoncé à se déterminer. 