fix tokenizer not support return_tensor='ms' (#2165)

Author: lvyufeng

README.md

@@ -70,7 +70,7 @@
     tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased")
     model = AutoModel.from_pretrained("bert-base-uncased")
 
-    inputs = tokenizer("Hello world!")
+    inputs = tokenizer("Hello world!", return_tensors='ms')
     outputs = model(**inputs)
     ```
 

mindnlp/core/npu/__init__.py

@@ -43,7 +43,7 @@ def device_count():
         return 0
     if GlobalComm.INITED:
         return get_group_size()
-    return ms_device_count()
+    return 1
 
 def current_device():
     return core.device('npu', 0)

mindnlp/transformers/__init__.py

@@ -56,6 +56,9 @@ def empty_fn(*args, **kwargs):
 transformers.tokenization_utils_base.PreTrainedTokenizerBase.apply_chat_template = apply_chat_template_wrapper(
     transformers.tokenization_utils_base.PreTrainedTokenizerBase.apply_chat_template
 )
+transformers.tokenization_utils_base.PreTrainedTokenizerBase.__call__ = apply_chat_template_wrapper(
+    transformers.tokenization_utils_base.PreTrainedTokenizerBase.__call__
+)
 
 transformers.pipelines.pipeline = dtype_wrapper(transformers.pipelines.pipeline)
 transformers.modeling_utils.caching_allocator_warmup = empty_fn