Update flash_dmattn/integrations/modeling_flash_dynamic_mask_attention_utils.py

LoserCheems · Copilot · web-flow · commit 587f0a6142bd · 2025-10-11T10:24:26.000+08:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/flash_dmattn/integrations/modeling_flash_dynamic_mask_attention_utils.py b/flash_dmattn/integrations/modeling_flash_dynamic_mask_attention_utils.py
@@ -461,7 +461,7 @@ def _process_flash_dynamic_mask_attention_kwargs(
 
     if supports_mapping["deterministic"]:
         flash_kwargs["deterministic"] = (
-            deterministic if deterministic is not None else os.getenv("FLASH_ATTENTION_DETERMINISTIC", "0") == "1"
+            deterministic if deterministic is not None else os.getenv("FLASH_DMATTN_DETERMINISTIC", "0") == "1"
         )
 
     if supports_mapping["softcap"] and softcap is not None:

Original file line number	Diff line number	Diff line change
`@@ -461,7 +461,7 @@ def _process_flash_dynamic_mask_attention_kwargs(`
`461`	`461`
`462`	`462`	`if supports_mapping["deterministic"]:`
`463`	`463`	`flash_kwargs["deterministic"] = (`
`464`		`- deterministic if deterministic is not None else os.getenv("FLASH_ATTENTION_DETERMINISTIC", "0") == "1"`
	`464`	`+ deterministic if deterministic is not None else os.getenv("FLASH_DMATTN_DETERMINISTIC", "0") == "1"`
`465`	`465`	`)`
`466`	`466`
`467`	`467`	`if supports_mapping["softcap"] and softcap is not None:`