We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fba0642 commit 218ff20Copy full SHA for 218ff20
tests/runai_model_streamer_test/test_weight_utils.py
@@ -23,10 +23,11 @@ def test_runai_model_loader():
23
runai_model_streamer_tensors = {}
24
hf_safetensors_tensors = {}
25
26
- for name, tensor in runai_safetensors_weights_iterator(safetensors):
+ for name, tensor in runai_safetensors_weights_iterator(safetensors,
27
+ True):
28
runai_model_streamer_tensors[name] = tensor
29
- for name, tensor in safetensors_weights_iterator(safetensors):
30
+ for name, tensor in safetensors_weights_iterator(safetensors, True):
31
hf_safetensors_tensors[name] = tensor
32
33
assert len(runai_model_streamer_tensors) == len(hf_safetensors_tensors)
0 commit comments