Browse Source

fix: typo in llama modeling file

AlpinDale 1 year ago
parent
commit
0715cc1958
1 changed files with 1 additions and 1 deletions
  1. 1 1
      aphrodite/modeling/models/llama.py

+ 1 - 1
aphrodite/modeling/models/llama.py

@@ -276,7 +276,7 @@ class LlamaForCausalLM(nn.Module):
 
 
             param = state_dict[name]
-            load_tensor_parallel_weights(param, loaded_weights, name,
+            load_tensor_parallel_weights(param, loaded_weight, name,
                                         self._column_parallel_weights,
                                         self._row_parallel_weights,
                                         tensor_model_parallel_rank)