Modalities / modalities

A framework for training multimodal foundation models.

MIT License

57 stars 5 forks source link

Fix CPU Tests with Github Actions for torch>=2.4 #205

Closed flxst closed 1 month ago

flxst commented 1 month ago

What does this PR do?

This PR fixes #203.

The problem was caused by a breaking change in torch 2.4 regarding the way caching of device_count works, compare https://pytorch.org/docs/2.3/_modules/torch/cuda.html#device_count and https://pytorch.org/docs/2.4/_modules/torch/cuda.html#device_count

General Changes

none except for the above

Breaking Changes

none

Checklist before submitting final PR

[x] My PR is minimal and addresses one issue in isolation
[x] I have merged the latest version of the target branch into this feature branch
[x] I have reviewed my own code w.r.t. correct implementation, missing type hints, proper documentation, etc.
[ ] I have run a sample config for model training
[x] I have checked that all tests run through (python tests/tests.py)