This commit is contained in:
shixian.shi 2024-01-12 18:02:10 +08:00
parent bcb8b0c3cb
commit 09a28d19df

View File

@ -168,7 +168,6 @@ class AutoModel:
device = "cpu"
# kwargs["batch_size"] = 1
kwargs["device"] = device
import pdb; pdb.set_trace()
if kwargs.get("ncpu", None):
torch.set_num_threads(kwargs.get("ncpu"))