Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
TST: Potentially Skip 8bit bnb regression test if compute capability …
…is too low (huggingface#1998) * TST Potentially Skip 8bit bnb regression test The 8bit bnb LoRA regression test results are dependent on the underlying compute capability. The logits are slightly different depending on the version (up to 0.5 abs diff). Therefore, we now check the compute capability for this test and skip it if it's too low. This check may require updating if the hardware of the CI worker is updated. Note that I have already invalidated the old regression artifacts and created a new one. * Fix pytest skip to work without cuda * Instead of skipping, add a comment to explain After internal discussion, we think this is the most practical solution for the time being.
- Loading branch information