a  d@s,ddlZddlmZdeeedddZdS) N)adaptive_clip_gradnorm@)valuemode norm_typecCsh|dkrtjjj|||dnF|dkr8tjj||n,|dkrPt|||dndsdJd|ddS) a Dispatch to gradient clipping method Args: parameters (Iterable): model parameters to clip value (float): clipping value/factor/norm, mode dependant mode (str): clipping mode, one of 'norm', 'value', 'agc' norm_type (float): p-norm, default 2.0 r)rrZagcFzUnknown clip mode (z).N)torchnnutilsZclip_grad_norm_Zclip_grad_value_r) parametersrrrr ]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/timm/utils/clip_grad.pydispatch_clip_grads r)rr)rZtimm.utils.agcrfloatstrrr r r r s