Inputs
Note:
cfg_trunc and renorm_cfg are advanced parameters. Renormalization only takes effect when renorm_cfg is greater than 0.0 and the current timestep is below cfg_trunc; if the new prediction norm is already below the computed maximum, no rescaling is performed.
Outputs
This documentation was AI-generated. If you find any errors or have suggestions for improvement, please feel free to contribute! Edit on GitHub
Source fingerprint (SHA-256):
5925bdfe2d62ef7261d73cda661834102ae6600b1afe53f4093568a6e83ec2ab