From a675161e47c3c1ccf893482642337281a7265d6c Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Mon, 10 Mar 2025 06:20:46 +0530 Subject: [PATCH] Fix nvidia 50xx support --- scripts/check_modules.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/check_modules.py b/scripts/check_modules.py index de8534be..448ea4ad 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -35,7 +35,7 @@ modules_to_check = { # "xformers": "0.0.16", "huggingface-hub": "0.21.4", "wandb": "0.17.2", - "torchruntime": "1.15.0", + "torchruntime": "1.15.1", "torchsde": "0.2.6", } modules_to_log = ["torchruntime", "torch", "torchvision", "sdkit", "stable-diffusion-sdkit", "diffusers"]