mirror of
https://github.com/storytold/FineTrainers-Conditioning.git
synced 2026-10-27 08:47:10 +00:00
bulk update styling and formatting (#170)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user