Skip to content

Commit

Permalink
doc fix for constrastive sampler (#1488)
Browse files Browse the repository at this point in the history
Fixes #1481
  • Loading branch information
mattdangerw authored Mar 5, 2024
1 parent bde20e9 commit 194dbbd
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion keras_nlp/samplers/contrastive_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}}
Expand Down

0 comments on commit 194dbbd

Please sign in to comment.