Skip to content

chatterbox(): add tune_gc opt-out for CUDA GC tuning#22

Merged
TroyHernandez merged 1 commit into
mainfrom
tune-gc-param
Jun 17, 2026
Merged

chatterbox(): add tune_gc opt-out for CUDA GC tuning#22
TroyHernandez merged 1 commit into
mainfrom
tune-gc-param

Conversation

@TroyHernandez

Copy link
Copy Markdown
Contributor

Follow-up to the CUDA GC tuning merged in #21 (0.1.0.15).

That tuning (.set_cuda_gc_options) sets persistent session-global torch.* options — deliberately, since torch reads the allocator rates later at CUDA init. The CRAN cookbook flags persistent options() changes, so this exposes the behavior as an opt-out instead of an unconditional side effect:

  • chatterbox() gains tune_gc = TRUE. Pass tune_gc = FALSE to skip the tuning.
  • @details documents the persistent side effect, the only-if-unset behavior, and the CUDA-only gating.

No behavior change at the default. This is the Option A approach (documented param) we discussed for chatterbox; whisper went with the fully opt-in whisper_tune_gc() function (cornball-ai/whisper#10).

Version 0.1.0.15 -> 0.1.0.16.

The CUDA GC tuning added in 0.1.0.15 (.set_cuda_gc_options) is a
persistent options() side effect. Expose it as a documented tune_gc
parameter (default TRUE) so callers can opt out with tune_gc = FALSE;
the Details section documents the side effect. No behavior change at the
default. Bumps to 0.1.0.16.
@TroyHernandez TroyHernandez merged commit 4b74757 into main Jun 17, 2026
4 checks passed
@TroyHernandez TroyHernandez deleted the tune-gc-param branch June 17, 2026 14:15
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