diff --git a/requirements.txt b/requirements.txt index 7f5a952..1d6146c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,7 +23,7 @@ triton>=2.1.0 # Experiment Tracking / Logging wandb>=0.15.0 -PyYAML>=6.0 +PyYAML>=6.0.3 tqdm>=4.65.0 loguru>=0.7.0 diff --git a/testing_requirements.txt b/testing_requirements.txt index a3969a8..36614e3 100644 --- a/testing_requirements.txt +++ b/testing_requirements.txt @@ -21,7 +21,7 @@ bitsandbytes>=0.41.1 deepspeed>=0.11.0 # Experiment Tracking / Logging wandb>=0.15.0 -PyYAML>=6.0 +PyYAML>=6.0.3 tqdm>=4.65.0 loguru>=0.7.0