Update BLOOM parameter counts (#18531)

* Update BLOOM parameter counts * Update BLOOM parameter counts
2022-08-12 19:36:18 +02:00
parent 153d1361c7
commit 56ef0ba447
6 changed files with 39 additions and 39 deletions
--- a/tests/onnx/test_onnx_v2.py
+++ b/tests/onnx/test_onnx_v2.py
@@ -215,7 +215,7 @@ PYTORCH_EXPORT_MODELS = {
 }

 PYTORCH_EXPORT_WITH_PAST_MODELS = {
-    ("bloom", "bigscience/bloom-350m"),
+    ("bloom", "bigscience/bloom-560m"),
    ("gpt2", "gpt2"),
    ("gpt-neo", "EleutherAI/gpt-neo-125M"),
 }