fix: default distributions to automatic GPU support with CPU fallback
This commit is contained in:
1 parent
d9336db1d5
commit
0fcc8ec21c
9 files changed
+43
-15
No files matched your search
@@ -17,7 +17,7 @@ def run(arguments, cwd=None):
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Build Searchi's pinned Google embedding worker")
|
||||
parser.add_argument("--backend", choices=["cpu", "auto", "metal", "vulkan", "cuda"], default="cpu")
|
||||
parser.add_argument("--backend", choices=["cpu", "auto", "metal", "vulkan", "cuda"], default="auto")
|
||||
parser.add_argument("--jobs", type=int, default=2)
|
||||
args = parser.parse_args()
|
||||
if not 1 <= args.jobs <= 16:
|
||||
|
||||
Reference in new issue
Block a user