diff --git a/fp8_requirements.txt b/fp8_requirements.txt index 198486004..17637f27e 100644 --- a/fp8_requirements.txt +++ b/fp8_requirements.txt @@ -1,5 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/nightly/cu121 -torch>=2.4.0.dev20240531,<2.4.1 +torch>=2.4.0 accelerate black==24.4.2 codeshield