bulk update styling and formatting (#170)

This commit is contained in:
Sayak Paul
2025-01-02 13:20:33 +05:30
committed by GitHub
parent 191ea8b0cc
commit 018953ed86
12 changed files with 26 additions and 24 deletions
+1 -1
View File
@@ -32,4 +32,4 @@ def align_device_and_dtype(
def expand_tensor_to_dims(tensor, ndim):
while len(tensor.shape) < ndim:
tensor = tensor.unsqueeze(-1)
return tensor
return tensor