Update model.py

import subprocess (to shell out for nvidia-smi) and shutil (to duplicate available() functionality and test whether we're in podman or docker Signed-off-by: Fred N. Garvin, Esq. <[email protected]>
containers · Nov 22, 2024 · c638d97 · c638d97
1 parent 5a08cb7
commit c638d97
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ramalama/model.py b/ramalama/model.py
@@ -3,6 +3,8 @@
 import glob
 import atexit
 import shlex
+import subprocess
+import shutil
 
 from ramalama.common import (
     default_image,