@@ -12,11 +12,11 @@ install_requires = [
"Jinja2==3.1.4",
"netifaces==0.11.0",
"numpy==2.0.0",
+ "nvidia-ml-py==12.560.30",
"pillow==10.4.0",
"prometheus-client==0.20.0",
"protobuf==5.27.1",
"psutil==6.0.0",
- "pynvml==11.5.3",
"requests==2.32.3",
"rich==13.7.1",
"safetensors==0.4.3",