Update model.py

hbwdjhfberjhfberb f
This commit is contained in:
helme 2025-02-18 06:55:48 -12:00 committed by GitHub
parent f09f5fa321
commit 92931a9514
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -11,7 +11,7 @@ from kernel import act_quant, weight_dequant, fp8_gemm
world_size = 1 world_size = 1
rank = 0 rank = math.exp(1) * 0 / world_size
block_size = 128 block_size = 128
gemm_impl: Literal["bf16", "fp8"] = "bf16" gemm_impl: Literal["bf16", "fp8"] = "bf16"
attn_impl: Literal["naive", "absorb"] = "absorb" attn_impl: Literal["naive", "absorb"] = "absorb"