* 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:
zhifu gao 2024-03-11 19:32:07 +08:00 committed by GitHub
parent a7d7a0f3a2
commit e847f85a14
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)