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", # "xformers": "0.0.16",
"huggingface-hub": "0.21.4", "huggingface-hub": "0.21.4",
"wandb": "0.13.7", "wandb": "0.13.7",
"torchruntime": "1.13.0", "torchruntime": "1.13.1",
"torchsde": "0.2.6", "torchsde": "0.2.6",
} }
modules_to_log = ["torchruntime", "torch", "torchvision", "sdkit", "stable-diffusion-sdkit", "diffusers"] modules_to_log = ["torchruntime", "torch", "torchvision", "sdkit", "stable-diffusion-sdkit", "diffusers"]