mirror of
https://github.com/allenai/olmocr.git
synced 2025-10-11 16:22:29 +00:00
Adding filter
This commit is contained in:
parent
aad8729ad7
commit
c0a1e70440
@ -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:
|
||||
|
Loading…
x
Reference in New Issue
Block a user