Add the top N-sigma sampler (#6796)

This commit is contained in:
oobabooga 2025-03-14 16:45:11 -03:00 committed by GitHub
parent 677d74a6a0
commit 5bcd2d7ad0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 60 additions and 3 deletions

View file

@ -21,6 +21,7 @@ class GenerationOptions(BaseModel):
eta_cutoff: float = 0
tfs: float = 1
top_a: float = 0
top_n_sigma: float = 0
dry_multiplier: float = 0
dry_allowed_length: int = 2
dry_base: float = 1.75