From 68a14d7da9b9e08c3b337b96ae8a92d163fe4091 Mon Sep 17 00:00:00 2001 From: osmarks Date: Sun, 8 Oct 2023 22:54:06 +0100 Subject: [PATCH] unignore device --- clip_server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clip_server.py b/clip_server.py index c937a54..6a2ea7e 100644 --- a/clip_server.py +++ b/clip_server.py @@ -23,7 +23,7 @@ import numpy with open(sys.argv[1], "r") as config_file: CONFIG = json.load(config_file) -DEVICE = "cuda:0" +DEVICE = CONFIG["device"] # So400m/14@384 with init_empty_weights():