faster pytest

Author: rebel-thkim

tests/test_llm.py

@@ -545,7 +545,8 @@ def setUpClass(cls):
         text_config = json.loads(config.text_config.to_json_string())
         text_config["num_hidden_layers"] = 1
         text_config["layer_types"] = text_config["layer_types"][:1]
-        vision_config["depth"] = 1  # To make the test faster
+        vision_config["depth"] = 2  # To make the test faster
+        vision_config["fullatt_block_indexes"] = [1]
         kwargs = {"vision_config": vision_config, "text_config": text_config}
         cls.HF_CONFIG_KWARGS.update(kwargs)
         return super().setUpClass()
@@ -590,8 +591,8 @@ def setUpClass(cls):
         text_config = json.loads(config.text_config.to_json_string())
         text_config["num_hidden_layers"] = 1
         text_config["layer_types"] = text_config["layer_types"][:1]
-        vision_config["depth"] = 8
-        vision_config["fullatt_block_indexes"] = [7]
+        vision_config["depth"] = 2
+        vision_config["fullatt_block_indexes"] = [1]
         kwargs = {"vision_config": vision_config, "text_config": text_config}
         cls.HF_CONFIG_KWARGS.update(kwargs)
         return super().setUpClass()

tests/test_transformers.py

@@ -416,7 +416,8 @@ def setUpClass(cls):
         # Reduce model size for faster testing
         vision_config = json.loads(config.vlm_config.vision_config.to_json_string())
         text_config = json.loads(config.vlm_config.text_config.to_json_string())
-        vision_config["depth"] = 1
+        vision_config["depth"] = 2
+        vision_config["fullatt_block_indexes"] = [1]
         text_config["num_hidden_layers"] = 1
         text_config["layer_types"] = text_config["layer_types"][:1]