osmarks/meme-search-engine
1
0
Fork 0
mirror of https://github.com/osmarks/meme-search-engine.git synced 2025-06-08 09:24:05 +00:00
Code Issues Releases Wiki Activity

???

Browse Source
This commit is contained in:
osmarks 2025-03-26 12:33:26 +00:00
parent 9af7eed0f9
commit ea3375a7bf
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clip_server.py
View File

@ -169,12 +169,16 @@ async def run_inference(request):
print(results[1]) print(results[1])
return web.Response(body=msgpack.dumps(body_data), status=status, content_type="application/msgpack") 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") @routes.get("/config")
async def config(request): async def config(request):
return web.Response(body=msgpack.dumps({ return web.Response(body=msgpack.dumps({
"model": CONFIG["model"], "model": CONFIG["model"],
"batch": BS, "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 "embedding_size": model.text.text_projection.out_features
}), status=200, content_type="application/msgpack") }), status=200, content_type="application/msgpack")