Add option to skip model for install

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

BSD 3-Clause "New" or "Revised" License

854 stars 275 forks source link

Add option to skip model for install #2308

Closed eellison closed 3 months ago

eellison commented 3 months ago

We are having a CI issue currently because a model's requirements.txt upgrades numpy to 2.0. It would be nice to have an option to skip one particular model.