Skip to content

fix(checkpoints): constrain tqdm for Hugging Face downloads - #140

Merged
lfengad merged 1 commit into
mainfrom
liangf/fix-hf-tqdm-compat
Jul 28, 2026
Merged

fix(checkpoints): constrain tqdm for Hugging Face downloads#140
lfengad merged 1 commit into
mainfrom
liangf/fix-hf-tqdm-compat

Conversation

@lfengad

@lfengad lfengad commented Jul 28, 2026

Copy link
Copy Markdown
Collaborator

Outcome

Restore Cosmos3 checkpoint downloads after tqdm 4.70 changed concurrent iterable length handling. The isolated Xet-capable hf@1.16.4 tool now resolves with tqdm<4.70; the repository-bundled Hugging Face dependency remains unchanged.

Upstream context: https://github.com/tqdm/tqdm/releases/tag/v4.70.0

Review path

Review the two added uvx requirement arguments in cosmos_framework/utils/checkpoint_db.py. Existing checkpoint download behavior is otherwise unchanged.

Verification

  • Verified _hf_download constructs uvx --with click --with tqdm<4.70 hf@1.16.4 download ... using a mocked subprocess.
  • uv run ruff format --check cosmos_framework/utils/checkpoint_db.py
  • git diff --check
  • ruff check reports only the pre-existing import-order issue on origin/main in the same file.

Risk and rollback

The constraint applies only to the transient Hugging Face CLI environment. Rollback is removal of the added --with tqdm<4.70 requirement.

@lfengad
lfengad enabled auto-merge (squash) July 28, 2026 07:00
@lfengad
lfengad merged commit 57d250e into main Jul 28, 2026
9 checks passed
@lfengad
lfengad deleted the liangf/fix-hf-tqdm-compat branch July 28, 2026 07:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants