don't use remote code for phi anymore

axolotl-ai-cloud · Jan 23, 2024 · 4350b95 · 4350b95
1 parent 97e06f3
commit 4350b95
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/tests/e2e/patched/test_phi_multipack.py b/tests/e2e/patched/test_phi_multipack.py
@@ -30,7 +30,6 @@ def test_ft_packed(self, temp_dir):
         cfg = DictDefault(
             {
                 "base_model": "microsoft/phi-1_5",
-                "trust_remote_code": True,
                 "model_type": "PhiForCausalLM",
                 "tokenizer_type": "AutoTokenizer",
                 "sequence_len": 2048,
@@ -78,7 +77,6 @@ def test_qlora_packed(self, temp_dir):
         cfg = DictDefault(
             {
                 "base_model": "microsoft/phi-1_5",
-                "trust_remote_code": True,
                 "model_type": "PhiForCausalLM",
                 "tokenizer_type": "AutoTokenizer",
                 "sequence_len": 2048,

diff --git a/tests/e2e/test_phi.py b/tests/e2e/test_phi.py
@@ -30,7 +30,6 @@ def test_phi_ft(self, temp_dir):
         cfg = DictDefault(
             {
                 "base_model": "microsoft/phi-1_5",
-                "trust_remote_code": True,
                 "model_type": "AutoModelForCausalLM",
                 "tokenizer_type": "AutoTokenizer",
                 "sequence_len": 2048,
@@ -76,7 +75,6 @@ def test_phi_qlora(self, temp_dir):
         cfg = DictDefault(
             {
                 "base_model": "microsoft/phi-1_5",
-                "trust_remote_code": True,
                 "model_type": "AutoModelForCausalLM",
                 "tokenizer_type": "AutoTokenizer",
                 "sequence_len": 2048,