From 9690fd1fa868849f81b22f30dddcd78758a9b247 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 1 Aug 2023 12:45:56 +0530 Subject: [PATCH] sdkit 1.0.154 - restrict tensorrt from 768x768 to 1024x1024, and Unet-only, to avoid going out of memory --- 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 d5886a9a..6bf3b1b3 100644 --- a/scripts/check_modules.py +++ b/scripts/check_modules.py @@ -18,7 +18,7 @@ os_name = platform.system() modules_to_check = { "torch": ("1.11.0", "1.13.1", "2.0.0"), "torchvision": ("0.12.0", "0.14.1", "0.15.1"), - "sdkit": "1.0.153", + "sdkit": "1.0.154", "stable-diffusion-sdkit": "2.1.4", "rich": "12.6.0", "uvicorn": "0.19.0",