From ea3375a7bf6b606ca8a8b9aae009a866df911ec9 Mon Sep 17 00:00:00 2001 From: osmarks Date: Wed, 26 Mar 2025 12:33:26 +0000 Subject: [PATCH] ??? --- clip_server.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/clip_server.py b/clip_server.py index bc3bc04..b974b2b 100644 --- a/clip_server.py +++ b/clip_server.py @@ -169,12 +169,16 @@ async def run_inference(request): print(results[1]) return web.Response(body=msgpack.dumps(body_data), status=status, content_type="application/msgpack") +image_size = [ t for t in preprocess.transforms if isinstance(t, transforms.Resize) ][0].size +if not isinstance(image_size, tuple): + image_size = (image_size, image_size) + @routes.get("/config") async def config(request): return web.Response(body=msgpack.dumps({ "model": CONFIG["model"], "batch": BS, - "image_size": [ t for t in preprocess.transforms if isinstance(t, transforms.Resize) ][0].size, + "image_size": image_size, "embedding_size": model.text.text_projection.out_features }), status=200, content_type="application/msgpack")