diff --git a/python/sglang/srt/models/grok.py b/python/sglang/srt/models/grok.py index 1f4a3b443..ac5bff3dd 100644 --- a/python/sglang/srt/models/grok.py +++ b/python/sglang/srt/models/grok.py @@ -221,7 +221,7 @@ def get_rope_scaling(config): "attn_factor": attn_factor, "beta_fast": beta_fast, "beta_slow": beta_slow, - "dtype": torch.float, + "dtype": torch.bfloat16, } return rope_scaling else: