remove space

Author: iamzainhuda

torchtune/modules/transformer.py

@@ -713,5 +713,4 @@ def unembed(self, h):
             # shape: [b, seq_len, out_dim]
             output = self.output(h).float()
 
-        return output
-    
+        return output