yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-04 12:07:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
Jake Poznanski 2024-10-07 14:32:47 -07:00
parent 7416b42023
commit fdcd77eadd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pdelfin/train/train.py
View File

@ -116,7 +116,7 @@ def run_train(config: TrainConfig):
setup_environment(aws_config=config.aws, wandb_config=config.wandb, WANDB_RUN_GROUP=run_name.group) setup_environment(aws_config=config.aws, wandb_config=config.wandb, WANDB_RUN_GROUP=run_name.group)
processor = AutoProcessor.from_pretrained(config.model.name_or_path) processor = AutoProcessor.from_pretrained(config.model.name_or_path)
train_dataset, valid_dataset = make_dataset(config) train_dataset, valid_dataset = make_dataset(config, processor)
model = Qwen2VLForConditionalGeneration.from_pretrained( model = Qwen2VLForConditionalGeneration.from_pretrained(
config.model.name_or_path, torch_dtype=torch.bfloat16, config.model.name_or_path, torch_dtype=torch.bfloat16,