a 8!ds@sddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZddlmZmZGdddZGd d d eZGd d d eZGd ddeZGdddeZGdddeZdS)N) PNDMScheduler DDIMSchedulerLMSDiscreteSchedulerEulerDiscreteSchedulerEulerAncestralDiscreteSchedulerDPMSolverMultistepScheduler)logger)DiffusionInpaintModel)torch_gcset_seed)Config SDSamplerc@s(eZdZddZddZeddZdS)CPUTextEncoderWrappercCsD|j|_|jtddd|_|jjtjdd|_||_~tdS)NcpuT) non_blocking)configtotorchdevice text_encoderfloat32 torch_dtyper )selfrrr?D:\my_projects\my_pycharmprojects\lama-cleaner-demo\model\sd.py__init__s zCPUTextEncoderWrapper.__init__cKs6|j}|j||jjfi|d||jgS)Nr)rrrr)rxkwargs input_devicerrr__call__s zCPUTextEncoderWrapper.__call__cCs|jS)N)r)rrrrdtype'szCPUTextEncoderWrapper.dtypeN)__name__ __module__ __qualname__rrpropertyr rrrrrsrc@sLeZdZdZdZejdddZedddZ d d Z e e d d d Z dS)SDi)rcKsNddlm}|dd }d|d|di}|dsB|ddr`td |tdddd |td kovtj }|r|rtj ntj }|j |jf|r|rd nd ||dd||_|j|ddr|j|ddr|rtd|jjddn4|j||_|drs   u