Sfoglia il codice sorgente

fix: typo in llama modeling file

AlpinDale 1 anno fa
parent
commit
0715cc1958
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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)