Skip to content

fix: make DiT checkpoint path configurable in kernel_launch_counter - #1

Open
heroarmor wants to merge 1 commit into
update-to-consolidated-kernelsfrom
fix/dit-ckpt-path-configurable
Open

fix: make DiT checkpoint path configurable in kernel_launch_counter#1
heroarmor wants to merge 1 commit into
update-to-consolidated-kernelsfrom
fix/dit-ckpt-path-configurable

Conversation

@heroarmor

Copy link
Copy Markdown
Owner

Problem

tools/kernel_launch_counter.py hardcoded the --ckpt default to an absolute path (/gpfs/.../zhkangqi/scmp_llm/pretrained_models/DiT-XL-2-256x256.pt) that no longer exists (the scmp_llm directory is gone), so the tool failed out of the box and wasn't portable.

Fix

Read the checkpoint path from $DIT_CKPT with a repo-relative fallback (pretrained_models/DiT-XL-2-256x256.pt).

🤖 Generated with Claude Code

The --ckpt default was a hardcoded absolute path
(/gpfs/.../zhkangqi/scmp_llm/pretrained_models/DiT-XL-2-256x256.pt) that no
longer exists (the scmp_llm dir is gone), so the tool failed out of the box and
wasn't portable across users. Read it from $DIT_CKPT with a repo-relative
fallback instead.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
heroarmor pushed a commit that referenced this pull request Aug 18, 2026
Bootstrap scmp_diffusion from scmp_llm
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.

2 participants