From 2be398c8524f1efcdbaf59694a0e382e59458b4a Mon Sep 17 00:00:00 2001 From: Umesh Chand Date: Thu, 11 Jun 2026 22:39:53 +0000 Subject: [PATCH] temprature as numpy.float64 is captured as a fake tensor during torch.export --- .../speech_transformer/transformer/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchbenchmark/models/speech_transformer/speech_transformer/transformer/attention.py b/torchbenchmark/models/speech_transformer/speech_transformer/transformer/attention.py index 4be85526df..151f808de4 100644 --- a/torchbenchmark/models/speech_transformer/speech_transformer/transformer/attention.py +++ b/torchbenchmark/models/speech_transformer/speech_transformer/transformer/attention.py @@ -21,7 +21,7 @@ def __init__(self, n_head, d_model, d_k, d_v, dropout=0.1): nn.init.normal_(self.w_vs.weight, mean=0, std=np.sqrt(2.0 / (d_model + d_v))) self.attention = ScaledDotProductAttention( - temperature=np.power(d_k, 0.5), attn_dropout=dropout + temperature=float(np.power(d_k, 0.5)), attn_dropout=dropout ) self.layer_norm = nn.LayerNorm(d_model)