New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update to latest ai-toolkit, torch, and cuda #27

Merged

andreasjansson merged 1 commit into main from upgrade-ai-toolkit-torch

Sep 7, 2024

Member

andreasjansson commented Sep 6, 2024 •

edited by ellipsis-dev bot

Loading

There are (maybe) some performance improvements to doing this, and some added features I'll use in later PRs.

🚀	This description was created by Ellipsis for commit `d077ad0`

Summary:

Updated ai-toolkit, torch, torchvision, and cuda versions in cog.yaml for potential performance improvements and future feature use.

Key points:

Updated ai-toolkit submodule to commit 279ee65177a1ff7110fc05d6c2b7e082787ea8bc.
Updated cuda version from 12.1 to 12.4 in cog.yaml.
Updated torch from 2.3.0 to 2.4.0 and torchvision from 0.18.0 to 0.19.0 in cog.yaml.
Added a command to force reinstall nightly versions of torch and torchvision with cuda 12.4 support in cog.yaml.

Generated with ❤️ by ellipsis.dev


          Update to latest ai-toolkit, torch, and cuda

d077ad0

andreasjansson requested review from fofr and zsxkib

September 6, 2024 19:55

andreasjansson mentioned this pull request

Train specific layers #28

Merged

andreasjansson merged commit 18c4a1c into main

2 of 4 checks passed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet