From c0a1e704400501ea10de13930c6a77562efd94fc Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Mon, 8 Sep 2025 17:02:09 +0000 Subject: [PATCH] Adding filter --- olmocr/train/prepare_checkpoint.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/olmocr/train/prepare_checkpoint.py b/olmocr/train/prepare_checkpoint.py index e8b6443..7257a1e 100755 --- a/olmocr/train/prepare_checkpoint.py +++ b/olmocr/train/prepare_checkpoint.py @@ -258,6 +258,9 @@ def prepare_checkpoints(sources: list[str], dest_path: str) -> None: """Prepare OlmOCR checkpoint(s) for deployment, with support for souping.""" print(f"Preparing {'souped ' if len(sources) > 1 else ''}checkpoint from {len(sources)} source(s) to {dest_path}") + sources = [source.rstrip("/") for source in sources] + dest_path = dest_path.rstrip("/") + # Detect architectures architectures = [] for source in sources: