osmarks/meme-search-engine
1
0
Fork 0
mirror of https://github.com/osmarks/meme-search-engine.git synced 2026-09-10 19:35:47 +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
Download Patch File Download Diff File Expand all files Collapse all files
clip_server.py
+1 -1
View File
@@ -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():