Skip to content

Fix module-level import with HardNegCollator#62

Merged
tonywu71 merged 3 commits into
mainfrom
fix-import-issue-with-HardNegCollator
Sep 12, 2024
Merged

Fix module-level import with HardNegCollator#62
tonywu71 merged 3 commits into
mainfrom
fix-import-issue-with-HardNegCollator

Conversation

@tonywu71
Copy link
Copy Markdown
Contributor

@tonywu71 tonywu71 commented Sep 12, 2024

Features

Fixed

  • Import HardNegCollator at module-level if and only if datasets is available

@tonywu71 tonywu71 self-assigned this Sep 12, 2024
@tonywu71 tonywu71 merged commit 4b38017 into main Sep 12, 2024
@tonywu71 tonywu71 deleted the fix-import-issue-with-HardNegCollator branch September 12, 2024 11:55
tonywu71 added a commit to tonywu71/colpali that referenced this pull request May 27, 2026
- Add bench_lik/bench_config.yaml: tiny ColQwen2 training config (B=16,
  30 steps, LoRA r=32, bf16, H100:1) for kernel A/B
- Add bench_lik/bench_train.py: runner capturing per-step time, loss, and
  peak VRAM; writes structured JSON
- Add bench_lik/sky_bench_lik.yaml: SkyPilot launch spec, pins LIK to
  main@6151c44 (Lq-bucketing illuin-tech#62 + SMEM model fix illuin-tech#73)
- Add bench_lik/render_report.py: HTML report renderer from result JSONs
- Add bench_lik/results/: post27 baseline + 6151c44 A/B result pairs
- Ignore uv.lock in .gitignore
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.

1 participant