diff --git a/requirements.txt b/requirements.txt index 6a4ba83..889b763 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,4 +4,5 @@ torch transformers datasets peft + pynvml>=11.5.0 \ No newline at end of file