a d@s ddlZddlmZddZdS)N)suppresscCs2|dkrtjjjS|dks"|dkr*ddStSdS)NampZ amp_bfloat16Zamp_bf16cSstjjjtjdS)N)Zdtype)torchcudarautocastZbfloat16rr[/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/training/precision.py zget_autocast..)rrrrr)Z precisionrrr get_autocasts  r )r contextlibrr rrrrs