LCM / utils /nn /init.py
shivrajanand's picture
Upload folder using huggingface_hub
e8f4897 verified
def assign_tensor(tensor, val):
"""
copy val to tensor
Args:
tensor: an n-dimensional torch.Tensor
val: an n-dimensional torch.Tensor to fill the tensor with
Returns:
"""
#if isinstance(tensor, Variable):
# assign_tensor(tensor.data, val)
# return tensor
return tensor.data.copy_(val)