Skip to content

Commit 77250e8

Browse files
committed
Increase GPU default mem alloc
1 parent 64349d9 commit 77250e8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

llama-tornado

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ def create_parser() -> argparse.ArgumentParser:
410410
const=Backend.PTX,
411411
help="Use PTX/CUDA backend",
412412
)
413-
hw_group.add_argument("--gpu-memory", default="7GB", help="GPU memory allocation")
413+
hw_group.add_argument("--gpu-memory", default="14GB", help="GPU memory allocation")
414414
hw_group.add_argument("--heap-min", default="20g", help="Minimum JVM heap size")
415415
hw_group.add_argument("--heap-max", default="20g", help="Maximum JVM heap size")
416416

0 commit comments

Comments
 (0)