Update peft version (mosaicml#3287)

mvpatel2000 · May 15, 2024 · 4806293 · 4806293
1 parent 01eec3a
commit 4806293
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -185,7 +185,7 @@ def package_files(prefix: str, directory: str, extension: str):
 ]
 
 extra_deps['peft'] = [
-    'peft>=0.7.0,<0.8',
+    'peft>=0.10.0,<0.11',
 ]
 
 extra_deps['sentencepiece'] = [

diff --git a/tests/loggers/test_mlflow_logger.py b/tests/loggers/test_mlflow_logger.py
@@ -416,6 +416,7 @@ def test_mlflow_save_model(tmp_path, tiny_gpt2_model, tiny_gpt2_tokenizer):
 
 @pytest.mark.filterwarnings('ignore:.*Setuptools is replacing distutils.*:UserWarning')
 @pytest.mark.filterwarnings("ignore:.*The 'transformers' MLflow Models integration.*:FutureWarning")
+@pytest.mark.filterwarnings('ignore:.*Could not find a config file.*:UserWarning')
 def test_mlflow_save_peft_model(tmp_path, tiny_mpt_model, tiny_mpt_tokenizer):
     mlflow = pytest.importorskip('mlflow')
     peft = pytest.importorskip('peft')