a d3@sRdZddlZddlZddlmZeeZejddddZ ejddddZ dS)z*Helper functions to detect NaN/Inf values.N)modelreturncCsF|D]8}|jdurt|jrt|d|jqdS)zVIterates over model parameters and prints out parameter + gradient information if NaN.Nz, ) parametersZgradtorchisnanfloatanyloginfo)rparamr r/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pytorch_lightning/utilities/finite_checks.pyprint_nan_gradientss rcCs<|D].\}}t|st|td|dqdS)zIterates over model parameters and prints gradients if any parameter is not finite. Raises: ValueError: If ``NaN`` or ``inf`` values are found z#Detected nan and/or inf values in `z?`. Check your forward pass for numerically unstable operations.N)Znamed_parametersrisfiniteallr ValueError)rnamer r r r detect_nan_parameterss  r) __doc__loggingrZtorch.nnnn getLogger__name__r Modulerrr r r r s