Skip to content

Commit

Permalink
use float arg
Browse files Browse the repository at this point in the history
  • Loading branch information
carlosgjs committed Jan 11, 2024
1 parent 9105f98 commit 70e4031
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/autora/doc/runtime/predict_hf.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def predict(
sys: str,
instr: str,
inputs: List[str],
do_sample: bool = False,
do_sample: float = 0.0,
temperature: float = 0.01,
top_p: float = 0.95,
top_k: float = 1,
Expand Down

0 comments on commit 70e4031

Please sign in to comment.