MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / get_use_flash_attention_2

Function get_use_flash_attention_2

k_diffusion/models/flags.py:13–14  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

11
12
13def get_use_flash_attention_2():
14 return os.environ.get("K_DIFFUSION_USE_FLASH_2", "1") == "1"
15
16
17state = threading.local()

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected