Skip to content

load compressed weights#47

Open
technillogue wants to merge 4 commits intomainfrom
syl/nyacomp
Open

load compressed weights#47
technillogue wants to merge 4 commits intomainfrom
syl/nyacomp

Conversation

@technillogue
Copy link
Contributor

@technillogue technillogue commented Jan 10, 2024

this uses https://github.com/technillogue/nyacomp to accelerate loading. it features nvCOMP bindings and tricks to start downloading and loading weights in parallel with python code. it can achieve load times as low as 9s in ideal conditions.

  • use nyacomp to load weights and bump libraries
  • include metadata files needed to load compressed
  • argument to compress or reload weights

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