fix

Signed-off-by: Onur Berk Töre <onurberk_t@hotmail.com>

Author: onurtore

prototype_source/nestedtensor.py

@@ -453,7 +453,7 @@ def zipf_sentence_lengths(alpha: float, batch_size: int) -> np.ndarray:
 
 # embeddings are assumed to be the same
 E = E_total
-mha_lib =  torch.nn.MultiheadAttention(E, nheads, batch_first=True, device=device)
+mha_lib = torch.nn.MultiheadAttention(E, nheads, batch_first=True, device=device)
 mha_lib.eval()
 
 ######################################################################