From a254c9f7bf3172aff8385174d761fa8bba508db0 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 6 Mar 2023 06:41:51 +0100 Subject: Update --- training/functional.py | 16 ---------------- 1 file changed, 16 deletions(-) (limited to 'training') diff --git a/training/functional.py b/training/functional.py index db46766..27a43c2 100644 --- a/training/functional.py +++ b/training/functional.py @@ -254,7 +254,6 @@ def loss_step( text_encoder: CLIPTextModel, with_prior_preservation: bool, prior_loss_weight: float, - perlin_strength: float, seed: int, step: int, batch: dict[str, Any], @@ -277,19 +276,6 @@ def loss_step( generator=generator ) - if perlin_strength != 0: - noise += perlin_strength * perlin_noise( - latents.shape[0], - latents.shape[1], - latents.shape[2], - latents.shape[3], - res=1, - octaves=4, - dtype=latents.dtype, - device=latents.device, - generator=generator - ) - # Sample a random timestep for each image timesteps = torch.randint( 0, @@ -574,7 +560,6 @@ def train( global_step_offset: int = 0, with_prior_preservation: bool = False, prior_loss_weight: float = 1.0, - perlin_strength: float = 0.1, **kwargs, ): text_encoder, unet, optimizer, train_dataloader, val_dataloader, lr_scheduler, extra = strategy.prepare( @@ -609,7 +594,6 @@ def train( text_encoder, with_prior_preservation, prior_loss_weight, - perlin_strength, seed, ) -- cgit v1.2.3-70-g09d2