Disable model check in UT (#411)

deepspeedai · Feb 12, 2024 · 47ce760 · 47ce760
1 parent af874da
commit 47ce760
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/legacy/conftest.py b/tests/legacy/conftest.py
@@ -106,13 +106,15 @@ def model_config(
     replace_with_kernel_inject: bool,
 ):
     config = SimpleNamespace(
+        skip_model_check=True, # TODO: remove this once conversation task check is fixed
         task=task_name,
         model=model_name,
         dtype=dtype,
         tensor_parallel=tensor_parallel,
         model_path=os.getenv("TRANSFORMERS_CACHE",
                              ""),
         meta_tensor=meta_tensor,
+        load_with_sys_mem=load_with_sys_mem,
         replica_num=replica_num,
         enable_deepspeed=enable_deepspeed,
         enable_zero=enable_zero,