a
    Ad_                     @   s|   d dl mZmZmZ d dlmZmZ g ejejZ	dd e	D Z
g ag ai Zi add Zdd Zd	d
 Zdd Ze  dS )    )sd_samplers_kdiffusionsd_samplers_timestepsshared)samples_to_image_gridsample_to_imagec                 C   s   i | ]}|j |qS  )name.0xr   r   ;/var/www/html/stable-diffusion-webui/modules/sd_samplers.py
<dictcomp>
       r   c                 C   s"   | d urt | d }ntd }|S )Nr   )all_samplers_mapgetall_samplers)r   configr   r   r   find_sampler_config   s    r   c                 C   sX   t | }|d usJ d|  |jrD|jddrDtd|j d||}||_|S )Nzbad sampler name: Zno_sdxlFzSampler z is not supported for SDXL)r   is_sdxloptionsr   	Exceptionr   constructorr   )r   modelr   samplerr   r   r   create_sampler   s    
r   c                  C   sT   t tjjatatat	  tD ].} | j
t| j
 < | jD ]}| j
t| < q:q d S )N)setr   optshide_samplerssamplers_hiddenr   samplerssamplers_for_img2imgsamplers_mapclearr   loweraliases)r   aliasr   r   r   set_samplers)   s    
r&   c                   C   s   dd t D S )Nc                 S   s   g | ]}|j tvr|j qS r   )r   r   r	   r   r   r   
<listcomp>8   r   z)visible_sampler_names.<locals>.<listcomp>)r   r   r   r   r   visible_sampler_names7   s    r(   N)modulesr   r   r   modules.sd_samplers_commonr   r   Zsamplers_data_k_diffusionZsamplers_data_timestepsr   r   r   r    r!   r   r   r   r&   r(   r   r   r   r   <module>   s    	