tiny-bird-diffusion / augment.py
sukriramli's picture
Upload augment.py with huggingface_hub
d44b839 verified
Raw
History Blame Contribute Delete
83 Bytes
import torch
def apply_white_noise(x):
return x + 0.005 * torch.randn_like(x)