diff --git a/keras_nlp/samplers/contrastive_sampler.py b/keras_nlp/samplers/contrastive_sampler.py index 4259167c8c..36d10690d7 100644 --- a/keras_nlp/samplers/contrastive_sampler.py +++ b/keras_nlp/samplers/contrastive_sampler.py @@ -34,7 +34,6 @@ class ContrastiveSampler(Sampler): alpha: float, the weight of minus max similarity in joint score computation. The larger the value of `alpha`, the score relies more on the similarity than the token probability. - seed: int. The random seed. Defaults to `None`. Call arguments: {{call_args}}