Skip to content

Commit d4f1f05

Browse files
waliwali777sevenan2
authored andcommitted
fix parallel_matmul
1 parent 7fd5643 commit d4f1f05

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

paddleformers/transformers/llama/modeling.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -328,8 +328,6 @@ class LlamaPretrainedModel(PretrainedModel):
328328

329329
@classmethod
330330
def _get_tensor_parallel_mappings(cls, config: LlamaConfig, is_split=True):
331-
if config.run_single_model:
332-
return {}
333331
from ..conversion_utils import split_or_merge_func
334332

335333
fn = split_or_merge_func(

0 commit comments

Comments
 (0)