Commit 17706785 authored by kurumuz's avatar kurumuz

fix

parent 6d18a731
......@@ -69,9 +69,9 @@ def sample_start_noise(seed, C, H, W, f, device="cuda"):
return noise
def sample_start_noise_special(seed, request, device="cuda"):
if request.seed is not None:
torch.manual_seed(request.seed)
np.random.seed(request.seed)
if seed:
torch.manual_seed(seed)
np.random.seed(seed)
noise = torch.randn([request.latent_channels, request.height // request.downsampling_factor, request.width // request.downsampling_factor], device=device).unsqueeze(0)
return noise
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment