make style

Author: JingyaHuang

optimum/neuron/utils/optimization_utils.py

@@ -85,7 +85,7 @@ def to3d(x):
     attention_scores = torch.bmm(key, query.transpose(-1, -2)) * (1 / math.sqrt(query.size(-1)))
     attention_probs = attention_scores.softmax(dim=1)
     if query.size() == key.size():
-       attention_probs = attention_probs.permute(0, 2, 1)
+        attention_probs = attention_probs.permute(0, 2, 1)
     attn_out = torch.bmm(attention_probs, value)
     if orig_shape:
         attn_out = attn_out.reshape(orig_shape[0], orig_shape[1], attn_out.shape[1], attn_out.shape[2])