Fix clm model with transformers 4.38.1 (#1324)

intel · Feb 28, 2024 · 98bfcf8 · 98bfcf8
1 parent 5b84e58
commit 98bfcf8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/huggingface/pytorch/language-modeling/quantization/requirements.txt b/examples/huggingface/pytorch/language-modeling/quantization/requirements.txt
@@ -8,5 +8,6 @@ pytest
 wandb
 einops
 neural-compressor
+pytest==8.0.0
 git+https://github.com/EleutherAI/lm-evaluation-harness.git@83dbfbf6070324f3e5872f63e49d49ff7ef4c9b3
 git+https://github.com/huggingface/peft.git@6c44096c7b8d55a2ecf24be9bc68393467e1584a