Skip to content

Commit

Permalink
Change min_p default to 1.0
Browse files Browse the repository at this point in the history
  • Loading branch information
oobabooga committed Nov 3, 2023
1 parent 367e5e6 commit 4537853
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/sampler_hijack.py
Original file line number Diff line number Diff line change
Expand Up @@ -252,8 +252,8 @@ def get_logits_processor_patch(self, **kwargs):

def generation_config_init_patch(self, **kwargs):
self.__init___old(**kwargs)
self.min_p = kwargs.pop("min_p", 1.0)
self.tfs = kwargs.pop("tfs", 1.0)
self.min_p = kwargs.pop("min_p", 0.0)
self.top_a = kwargs.pop("top_a", 0.0)
self.mirostat_mode = kwargs.pop("mirostat_mode", 0)
self.mirostat_eta = kwargs.pop("mirostat_eta", 0.1)
Expand Down

0 comments on commit 4537853

Please sign in to comment.