daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use configured H3 sampler capture directory

Browse source
This commit is contained in:
Daniel Maddern 2026-08-13 14:07:28 +07:00
parent 663534f5dc
commit 9d463b0201
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
tools/patch_comfy_sampler_capture.py
View file

@ -10,7 +10,7 @@ if "import os\n" not in source:
old = " if callback is not None:\n callback(x[\"i\"], x[\"denoised\"], x[\"x\"], total_steps)\n" old = " if callback is not None:\n callback(x[\"i\"], x[\"denoised\"], x[\"x\"], total_steps)\n"
new = ( new = (
" capture_dir = os.getenv(\"H3_SAMPLER_CAPTURE_DIR\")\n" " capture_dir = os.getenv(\"H3_SAMPLER_CAPTURE_DIR\") or os.getenv(\"H3_CAPTURE_DIR\")\n"
" if capture_dir:\n" " if capture_dir:\n"
" os.makedirs(capture_dir, exist_ok=True)\n" " os.makedirs(capture_dir, exist_ok=True)\n"
" torch.cuda.synchronize(x[\"x\"].device)\n" " torch.cuda.synchronize(x[\"x\"].device)\n"