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
containers · Nov 22, 2024 · 7e2bb6a · 7e2bb6a
1 parent 3c39592
commit 7e2bb6a
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,