o Û·iEã@sZddlZddlZddlmZddlmZmZmZddl m Z e   e ¡Z Gdd„deƒZdS)éN)Ú CLIPConfigÚCLIPVisionModelWithProjectionÚPreTrainedModelé)Úloggingcs>eZdZeZdgZdef‡fdd„ Ze ¡ddd„ƒZ ‡Z S) ÚIFSafetyCheckerÚCLIPEncoderLayerÚconfigcs@tƒ |¡t|jƒ|_t |jjd¡|_t |jjd¡|_ dS)Né) ÚsuperÚ__init__rÚ vision_configÚ vision_modelÚnnÚLinearÚprojection_dimÚp_headÚw_head)Úselfr ©Ú __class__©úe/home/ubuntu/vllm_env/lib/python3.10/site-packages/diffusers/pipelines/deepfloyd_if/safety_checker.pyr s  zIFSafetyChecker.__init__çà?c CsÔ| |¡d}| |¡}| ¡}||k}| ¡}t|ƒr!t d¡t|ƒD]\}}|r5t  ||j ¡||<q%|  |¡} |  ¡} | |k} |  ¡} t| ƒrPt d¡t| ƒD]\}} | rdt  ||j ¡||<qT||| fS)Nrz‘Potential NSFW content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed.z˜Potential watermarked content was detected in one or more images. A black image will be returned instead. Try again with a different prompt and/or seed.) rrÚflattenÚtolistÚanyÚloggerÚwarningÚ enumerateÚnpÚzerosÚshaper) rÚ clip_inputÚimagesÚ p_thresholdÚ w_thresholdÚ image_embedsÚ nsfw_detectedÚidxÚnsfw_detected_Úwatermark_detectedÚwatermark_detected_rrrÚforwards4 ÿ€ ÿ€ zIFSafetyChecker.forward)rr) Ú__name__Ú __module__Ú __qualname__rÚ config_classÚ_no_split_modulesr ÚtorchÚno_gradr-Ú __classcell__rrrrr s r)Únumpyr r3Útorch.nnrÚ transformersrrrÚutilsrÚ get_loggerr.rrrrrrÚs