update

microsoft · Feb 1, 2024 · 444386a · 444386a
1 parent 67661b9
commit 444386a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/onnxruntime/python/tools/transformers/models/llama/llama_parity.py b/onnxruntime/python/tools/transformers/models/llama/llama_parity.py
@@ -256,7 +256,7 @@ def main(argv: list[str] = []):  # noqa: B006
 
     kv_cache_ortvalues = {}
     if not args.merged:
-        verify_parity(args, location, kv_cache_ortvalues)
+        verify_parity(args, location, use_auth_token, kv_cache_ortvalues)
     else:
         # Verify prompt generation in merged model (decoder_model.onnx)
         args.use_past_kv = False

diff --git a/onnxruntime/python/tools/transformers/models/llama/requirements.txt b/onnxruntime/python/tools/transformers/models/llama/requirements.txt
@@ -1,6 +1,6 @@
 optimum>=1.14.1
 transformers>=4.33.2
-torch>=2.2.0.dev20230920
+torch>=2.2.0
 onnx>=1.14.0
 datasets>=2.8.0
-protobuf==3.20.2
+protobuf==3.20.2