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 43d2642b68
commit 55a31c77e6

View File

@ -35,7 +35,7 @@ modules_to_check = {
# "xformers": "0.0.16",
"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"]