Skip to content

Navigation Menu

Explore
By company size
By use case
By industry
View all solutions
Topics
- AI
- DevOps
- Security
- Software Development
- View all
Explore
- GitHub Sponsors
  Fund open source developers
- The ReadME Project
  GitHub community articles
Repositories
- Enterprise platform
  AI-powered developer platform
Available add-ons
Pricing

Search code, repositories, users, issues, pull requests...

Search

Clear

Search syntax tips

Provide feedback

We read every piece of feedback, and take your input very seriously.

Include my email address so I can be contacted

Saved searches

Use saved searches to filter your results more quickly

Name

Query

To see all available qualifiers, see our documentation.

You signed in with another tab or window. Reload to refresh your session. You signed out in another tab or window. Reload to refresh your session. You switched accounts on another tab or window. Reload to refresh your session.

Dismiss alert

invoke-ai / InvokeAI Public

Notifications You must be signed in to change notification settings
Fork 2.5k
Star 24.5k

Code
Issues 636
Pull requests 38
Actions
Projects
Wiki
Security
Insights

Additional navigation options

Code
Issues
Pull requests
Actions
Projects
Wiki
Security
Insights

Add `pytorch_cuda_alloc_conf` config to tune VRAM memory allocation#7673

Open

RyanJDick wants to merge 6 commits intoryan/tidy-entryinvoke-ai/InvokeAI:ryan/tidy-entryfrom ryan/vae-decode-meminvoke-ai/InvokeAI:ryan/vae-decode-memCopy head branch name to clipboard

+85-7

Conversation Commits6 (6)Checks Files changed

Commits

Commits on Feb 24, 2025

Add utils for configuring the torch CUDA allocator.
RyanJDick
committed
Add use_cuda_malloc config option.
RyanJDick
committed
Simplify is_torch_cuda_malloc_enabled() implementation and add unit tests.
RyanJDick
committed
Switch from use_cuda_malloc flag to a general pytorch_cuda_alloc_conf config field that allows full customization of the CUDA allocator.
RyanJDick
committed
Update low-vram docs with info abhout .
RyanJDick
committed
Mark test_configure_torch_cuda_allocator_raises_if_torch_is_already_imported() to only run if CUDA is available.
RyanJDick
committed

Footer

© 2025 GitHub, Inc.

Footer navigation

Terms
Privacy
Security
Status
Docs
Contact

You can’t perform that action at this time.