mirror of
https://github.com/modelscope/FunASR
synced 2025-09-15 14:48:36 +08:00
Dev gzf (#1468)
* qwenaudio qwenaudiochat * qwenaudio qwenaudiochat * whisper * whisper * llm * llm * llm * llm * llm * llm * llm * llm * export onnx * export onnx * export onnx * dingding * dingding * llm * doc * onnx * onnx
This commit is contained in:
parent
a7d7a0f3a2
commit
e847f85a14
@ -155,9 +155,9 @@ class AutoModel:
|
||||
device = "cpu"
|
||||
kwargs["batch_size"] = 1
|
||||
kwargs["device"] = device
|
||||
|
||||
if kwargs.get("ncpu", 4):
|
||||
torch.set_num_threads(kwargs.get("ncpu"))
|
||||
|
||||
torch.set_num_threads(kwargs.get("ncpu", 4))
|
||||
|
||||
|
||||
# build tokenizer
|
||||
tokenizer = kwargs.get("tokenizer", None)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user