Revert "Add import hint"

This reverts commit 9cedf18a97.

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
 
 

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