torchruntime upgrade - fix bug with multiple GPUs of the same model

This commit is contained in:
cmdr2 2025-03-04 11:27:17 +05:30
parent 7a383d7bc4
commit 80a4c6f295

View File

@ -35,7 +35,7 @@ modules_to_check = {
"onnxruntime": "1.19.2",
"huggingface-hub": "0.21.4",
"wandb": "0.13.7",
"torchruntime": "1.13.0",
"torchruntime": "1.13.1",
"torchsde": "0.2.6",
}
modules_to_log = ["torchruntime", "torch", "torchvision", "sdkit", "stable-diffusion-sdkit", "diffusers"]