simplify by removing duplicate base_model_config (#772)

2023-10-23 01:42:38 -04:00
parent 44c9d0151a
commit 2d8def68dc
39 changed files with 10 additions and 42 deletions
--- a/tests/e2e/test_fused_llama.py
+++ b/tests/e2e/test_fused_llama.py
@@ -31,7 +31,6 @@ class TestFusedLlama(unittest.TestCase):
        cfg = DictDefault(
            {
                "base_model": "JackFram/llama-68m",
-                "base_model_config": "JackFram/llama-68m",
                "flash_attention": True,
                "flash_attn_fuse_qkv": True,
                "flash_attn_fuse_mlp": True,