Split test config into multiple-lines

quic · Jun 26, 2024 · 4b89dc7 · 4b89dc7
1 parent 4bf37ea
commit 4b89dc7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_loader.py b/tests/test_loader.py
@@ -14,7 +14,11 @@
 from QEfficient import QEFFAutoModelForCausalLM, QEFFCommonLoader
 
 model_name_to_params_dict: Dict[str, Dict[str, Any]] = {
-    "gpt2": {"qeff_class": QEFFAutoModelForCausalLM, "hf_class": GPT2LMHeadModel, "prompt": "Equator is"},
+    "gpt2": {
+        "qeff_class": QEFFAutoModelForCausalLM,
+        "hf_class": GPT2LMHeadModel,
+        "prompt": "Equator is",
+    },
 }
 model_names = model_name_to_params_dict.keys()