[diffusion] doc: minor update docs (#13177)

This commit is contained in:
Mick
2025-11-21 14:35:29 +08:00
committed by GitHub
parent 8c212a2029
commit eda2f70033
23 changed files with 38 additions and 35 deletions

View File

@@ -450,8 +450,8 @@ def import_pynvml():
status without initializing CUDA context in the current process.
Historically, there are two packages that provide pynvml:
- `nvidia-ml-py` (https://pypi.org/project/nvidia-ml-py/): The official
wrapper. It is a dependency of sgl-diffusion, and is installed when users
install sgl-diffusion. It provides a Python module named `pynvml`.
wrapper. It is a dependency of sglang-diffusion, and is installed when users
install sglang-diffusion. It provides a Python module named `pynvml`.
- `pynvml` (https://pypi.org/project/pynvml/): An unofficial wrapper.
Prior to version 12.0, it also provides a Python module `pynvml`,
and therefore conflicts with the official one which is a standalone Python file.