update return_numpy=True

Author: DesmonDay

paddlenlp/experimental/transformers/chatglm/modeling.py

@@ -631,7 +631,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model
 

paddlenlp/experimental/transformers/gpt/modeling.py

@@ -494,7 +494,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model
 

paddlenlp/experimental/transformers/llama/modeling.py

@@ -1185,7 +1185,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model
 
@@ -1322,7 +1322,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model
 
@@ -1599,7 +1599,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
 
         return model

paddlenlp/experimental/transformers/opt/modeling.py

@@ -377,7 +377,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model
 

paddlenlp/experimental/transformers/qwen/modeling.py

@@ -426,7 +426,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *args, **kwargs):
         )
 
         model_path = os.path.dirname(resolved_archive_file)
-        state_dict = load_tp_checkpoint(model_path, cls, config)
+        state_dict = load_tp_checkpoint(model_path, cls, config, return_numpy=True)
         model.set_state_dict(state_dict)
         return model