From 38b858926d5bd0deee05f59de3baf6d41ef423df Mon Sep 17 00:00:00 2001 From: rockerBOO Date: Tue, 15 Jul 2025 19:54:41 -0400 Subject: [PATCH] Revert "Add import hint" This reverts commit 9cedf18a974cd272c260da5bbe8d4c585f31a301. --- library/train_util.py | 3 --- train_network.py | 6 +----- 2 files changed, 1 insertion(+), 8 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index 09c101a8..d7243e07 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -6029,9 +6029,6 @@ def get_noise_noisy_latents_and_timesteps( else: noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps) - - noise_scheduler.alphas_cumprod = noise_scheduler.alphas_cumprod.to("cpu") - return noise, noisy_latents, timesteps diff --git a/train_network.py b/train_network.py index 02995c95..2eabcdf3 100644 --- a/train_network.py +++ b/train_network.py @@ -45,11 +45,7 @@ from library.custom_train_functions import ( apply_masked_loss, ) from library.utils import setup_logging, add_logging_arguments - -try: - from wavelet_loss import WaveletLoss -except: - raise ImportError("WaveletLoss is not installed. Please install it with `pip install git+https://github.com/rockerBOO/wavelet-loss`") +from wavelet_loss import WaveletLoss setup_logging() import logging