Switch the default `pytest` call to run just CPU test cases

apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner

MIT License

448 stars 31 forks source link

Open luke-carlson opened 1 week ago

luke-carlson commented 1 week ago

Rather than having to remember pytest -m "not gpu", maybe its easier if we assume that users first run test cases on cpu

bdeanhardt commented 4 days ago

Ethan and I can take this one!