Skip to content

Commit

Permalink
Merge pull request #802 from containers/ngl-999
Browse files Browse the repository at this point in the history
If ngl is not specified
  • Loading branch information
rhatdan authored Feb 13, 2025
2 parents 0c5015a + 7a0218a commit 08c5e25
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions ramalama/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,6 +223,9 @@ def gpu_args(self, args, runner=False):
and os.path.exists("/dev/dri")
)
):
if args.ngl < 0:
args.ngl = 999

if runner:
gpu_args += ["--ngl"] # double dash
else:
Expand Down

0 comments on commit 08c5e25

Please sign in to comment.