Skip to content

Commit

Permalink
Change default (#8371) (#8372)
Browse files Browse the repository at this point in the history
Signed-off-by: smajumdar <[email protected]>
Co-authored-by: Somshubra Majumdar <[email protected]>
Signed-off-by: Pablo Garay <[email protected]>
  • Loading branch information
2 people authored and pablo-garay committed Mar 19, 2024
1 parent 4e21a75 commit 59f7aa0
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -486,7 +486,7 @@ class MultiTaskDecodingConfig:
# )

# beam decoding config
beam: AEDBeamInferConfig = field(default_factory=lambda: AEDBeamInferConfig(beam_size=4))
beam: AEDBeamInferConfig = field(default_factory=lambda: AEDBeamInferConfig(beam_size=1))

# can be used to change temperature for decoding
temperature: float = 1.0

0 comments on commit 59f7aa0

Please sign in to comment.