diff --git a/examples/qualcomm/oss_scripts/llama/llama.py b/examples/qualcomm/oss_scripts/llama/llama.py index 99f346eccbc..db533986119 100755 --- a/examples/qualcomm/oss_scripts/llama/llama.py +++ b/examples/qualcomm/oss_scripts/llama/llama.py @@ -591,7 +591,7 @@ def permute(w, heads): for llama_instance in llama_instance_list: llama_instance.load_state_dict( state_dict, - strict=False, + strict=True, assign=True, ) end_load_ts = time.time()