Fix peft llama test (#1059)

mosaicml · Apr 18, 2024 · 88215fd · 88215fd
1 parent e52936c
commit 88215fd
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/a_scripts/inference/test_convert_composer_to_hf.py b/tests/a_scripts/inference/test_convert_composer_to_hf.py
@@ -586,9 +586,7 @@ def test_huggingface_conversion_callback(
                 'flavor': 'peft',
                 'path': ANY,
                 'save_pretrained_dir': ANY,
-                'metadata': {
-                    'task': 'llm/v1/completions'
-                }
+                'metadata': {},
             }
         else:
             import numpy as np