Fix `torch.load` of transformers patch when enabling `low_cpu_mem_usage`#10

Open

lausannel wants to merge 1 commit intomainfrom

fix/low_mem_replace

Contributor

lausannel commented Sep 23, 2024 •

edited

Loading

For Transformers versions ≤ 4.36, apply the original patch.
For Transformers versions > 4.37, insert a comma to prevent the following error.


          fix: different replacements for different transformer versions

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

Labels

None yet