finally remove torchsde warning

This commit is contained in:
lvmin
2023-10-14 04:50:13 -07:00
parent 79b1a605a7
commit e27383c07f
2 changed files with 5 additions and 1 deletions
+1 -1
View File
@@ -1 +1 @@
version = '2.1.62' version = '2.1.63'
+4
View File
@@ -19,6 +19,7 @@ import fcbh.samplers
import fcbh.cli_args import fcbh.cli_args
import args_manager import args_manager
import modules.advanced_parameters as advanced_parameters import modules.advanced_parameters as advanced_parameters
import warnings
from fcbh.k_diffusion import utils from fcbh.k_diffusion import utils
from fcbh.k_diffusion.sampling import BrownianTreeNoiseSampler, trange from fcbh.k_diffusion.sampling import BrownianTreeNoiseSampler, trange
@@ -505,4 +506,7 @@ def patch_all():
fcbh.k_diffusion.external.DiscreteEpsDDPMDenoiser.forward = patched_discrete_eps_ddpm_denoiser_forward fcbh.k_diffusion.external.DiscreteEpsDDPMDenoiser.forward = patched_discrete_eps_ddpm_denoiser_forward
fcbh.model_base.SDXL.encode_adm = sdxl_encode_adm_patched fcbh.model_base.SDXL.encode_adm = sdxl_encode_adm_patched
fcbh.sd1_clip.ClipTokenWeightEncoder.encode_token_weights = encode_token_weights_patched_with_a1111_method fcbh.sd1_clip.ClipTokenWeightEncoder.encode_token_weights = encode_token_weights_patched_with_a1111_method
warnings.filterwarnings(action='ignore', module='torchsde')
return return