osmarks/meme-search-engine
1
0
Fork 0
mirror of https://github.com/osmarks/meme-search-engine.git synced 2024-11-10 22:09:54 +00:00
Code Issues Releases Wiki Activity

unignore device

Browse Source
This commit is contained in:
osmarks 2023-10-08 22:54:06 +01:00
parent 20fcc9317f
commit 68a14d7da9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clip_server.py
View File

@ -23,7 +23,7 @@ import numpy
with open(sys.argv[1], "r") as config_file: with open(sys.argv[1], "r") as config_file:
CONFIG = json.load(config_file) CONFIG = json.load(config_file)
DEVICE = "cuda:0" DEVICE = CONFIG["device"]
# So400m/14@384 # So400m/14@384
with init_empty_weights(): with init_empty_weights():