o ۷i#@sJddlZddlZddlZddlmZeddadad defddZ dS) N) disable_printignorealexnetcCs|dkr&tdur#ttjddaWdn1swYt}n-|dkrLtdurIttjddaWdn1sDwYt}nJd|t|||WdS1siwYdS)Nr)rvggFzunsupport net ) lowerlpips_loss_fn_alexrlpipsLPIPSlpips_loss_fn_vggtorchno_graditem)img0img1rloss_fnrM/home/ubuntu/vllm_env/lib/python3.10/site-packages/cache_dit/metrics/lpips.pycompute_lpips_imgs    $r)r) warningsr r utilsrfilterwarningsr r strrrrrrs