Skip to content

GitLab

Unverified Commit 4ea62c0e authored by Ning Xie's avatar Ning Xie Committed by GitHub
Browse files

[CI] add missing argument (#18694) 


Signed-off-by: default avatarAndy Xie <andy.xning@gmail.com>
parent 561b77a0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
tests/runai_model_streamer_test/test_weight_utils.py
View file @ 4ea62c0e
...@@ -23,10 +23,11 @@ def test_runai_model_loader(): ...@@ -23,10 +23,11 @@ def test_runai_model_loader():
runai_model_streamer_tensors = {} runai_model_streamer_tensors = {}
hf_safetensors_tensors = {} hf_safetensors_tensors = {}
for name, tensor in runai_safetensors_weights_iterator(safetensors): for name, tensor in runai_safetensors_weights_iterator(
safetensors, True):
runai_model_streamer_tensors[name] = tensor runai_model_streamer_tensors[name] = tensor
for name, tensor in safetensors_weights_iterator(safetensors): for name, tensor in safetensors_weights_iterator(safetensors, True):
hf_safetensors_tensors[name] = tensor hf_safetensors_tensors[name] = tensor
assert len(runai_model_streamer_tensors) == len(hf_safetensors_tensors) assert len(runai_model_streamer_tensors) == len(hf_safetensors_tensors)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment