clip_ instead

pull/32/head
aria1th 2023-01-17 17:18:43 +09:00
parent 328692830c
commit 72b7609c3f
1 changed files with 2 additions and 2 deletions

View File

@ -130,8 +130,8 @@ class PersonalizedBase(Dataset):
elif latent_sampling_method == "random":
if latent_sampling_std != -1:
assert latent_sampling_std > 0, f"Cannnot apply negative standard deviation {latent_sampling_std}"
print(f"Applying patch, changing std from {latent_dist.std} to {latent_sampling_std}...")
latent_dist.std = latent_sampling_std
print(f"Applying patch, clipping std from {torch.max(latent_dist.std).item()} to {latent_sampling_std}...")
latent_dist.std.clip_(latent_sampling_std)
entry = DatasetEntry(filename=path, filename_text=filename_text, latent_dist=latent_dist)
if not (self.tag_drop_out != 0 or self.shuffle_tags):