We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8f51d3b commit 1374fd2Copy full SHA for 1374fd2
tests/models/test_decoders.py
@@ -69,7 +69,9 @@
69
USE_MICRO_MODELS = os.environ.get("FMS_TEST_SHAPES_USE_MICRO_MODELS", "1") == "1"
70
USE_DISTRIBUTED = os.environ.get("FMS_TEST_SHAPES_DISTRIBUTED", "0") == "1"
71
TIMING = os.environ.get("TIMING", "")
72
-CUMULATIVE_TEST_TOKENS_PER_SEQUENCE = os.environ.get("FMS_TEST_SHAPES_CUMULATIVE_TEST_TOKENS_PER_SEQUENCE", "1024")
+CUMULATIVE_TEST_TOKENS_PER_SEQUENCE = os.environ.get(
73
+ "FMS_TEST_SHAPES_CUMULATIVE_TEST_TOKENS_PER_SEQUENCE", "1024"
74
+)
75
ATTN_TYPE = os.environ.get("FMS_TEST_SHAPES_ATTN_TYPE", "sdpa")
76
attention_map = {
77
"sdpa": "sdpa_causal",
0 commit comments