mirror of
https://github.com/allenai/olmocr.git
synced 2025-10-14 17:52:53 +00:00
fix
This commit is contained in:
parent
1dbb4332c0
commit
3c8410f22c
@ -406,7 +406,6 @@ def main():
|
|||||||
warmup_steps=10,
|
warmup_steps=10,
|
||||||
max_completion_length=3000,
|
max_completion_length=3000,
|
||||||
temperature=0.7,
|
temperature=0.7,
|
||||||
do_sample=True,
|
|
||||||
report_to=report_to,
|
report_to=report_to,
|
||||||
remove_unused_columns=False,
|
remove_unused_columns=False,
|
||||||
torch_dtype=torch.bfloat16,
|
torch_dtype=torch.bfloat16,
|
||||||
|
Loading…
x
Reference in New Issue
Block a user