diff --git a/common/sampling.py b/common/sampling.py index c8366b3d..eab2a4cc 100644 --- a/common/sampling.py +++ b/common/sampling.py @@ -373,6 +373,7 @@ def to_gen_params(self, **kwargs): "dry_base": self.dry_base, "dry_allowed_length": self.dry_allowed_length, "dry_sequence_breakers": self.dry_sequence_breakers, + "dry_range": self.dry_range, "repetition_decay": self.repetition_decay, "mirostat": self.mirostat_mode == 2, "mirostat_tau": self.mirostat_tau,