hololens/stable-diffusion-webui-depthmap-script
1
1�
2 ��f� � � � d dl Z d dlmZ d dlZd dlZ G d� dej � Z d dlmZ d dlm Z d dlm3Z4 d dl m Z d d l
mZ d5� Zd � Zd � Zd
� Zd� Zd� Zd� Zej, Zy# ed� d� Zd� Z e ej4 � j7 � � Z d a d� Z6d� Z d� Zd� Zd� Zd� Zd� Zd� Zd � Zd � Zej< ZY yxY w) � N)�datetimec � � e Zd ZdZdZy)� BackboneType� � N)�__name__�7__module__� __qualname__�WEBUI�8STANDALONE� � �DF:\3dpicmaker\stable-diffusion-webui-depthmap-script\src\backbone.pyr r 9 s � �
�E��Jr r )�10save_image)�torch_gc)�get_next_sequence_number)�wrap_gradio_gpu_call)� listfilesc �L � ddl m} t || � r|j | � S |S )Nr )�opts)�modules.sharedr �hasattr� __getattr__)�name�defaultr s r �get_optr s% � �'� �4�� ��#�#�D�)� )��r c �L � ddl m} t || � r|j | � S |S )zGet command line argumentr )�cmd_opts)r r r �__getattribute__)r r r s r � get_cmd_optr s% � �+� �8�T� "��,�,�T�2� 2��r c � � i } dD ]' }t d|z d� }|� t |d� }|��#|| |< �) dD ] }|| v s�t | | � | |< � | S )z"Parameters for depthmap generation��11boost_rmax� precision�no_half�marigold_ensembles�marigold_steps�depthmap_script_N)r& r' )r r �int)�ops�s�cs r �12gather_opsr- $ sw � ���_� �A��*�Q�.��5�A��y� ��4�(���}���A�� � :� %�A��C�x� �S��V���A�� %� �13r c �z � t dd� } | �t | � dk( r t dd� } | �t | � dkD sJ �| S )z+Get path where results are saved by default�outdir_samplesNr �outdir_extras_samples)r �len)�paths r � get_outpathr3 4 sC � ��'��.�� �<�3�t�9��>��2�D�9�D�� �C��I��M�1�� r c �D � ddl m} m} | j ��| j j �/| j j j |j � | j j �0| j j j |j � y y y �Nr ��shared�devices)�modulesr7 r8 �sd_model�cond_stage_model�to�cpu�first_stage_modelr6 s r �unload_sd_modelr? = sq � �+� �?�?� &����/�/�;����0�0�3�3�G�K�K�@����0�0�<����1�1�4�4�W�[�[�A� =� 'r c �D � ddl m} m} | j ��| j j �/| j j j |j � | j j r0| j j j |j � y y y r5 )r9 r7 r8 r: r; r< �devicer> r6 s r �reload_sd_modelrB G sn � �+� �?�?� &����/�/�;����0�0�3�3�G�N�N�C����0�0����1�1�4�4�W�^�^�D� 1� 'r c �6 � dd l } | j j S )Nr )r r7 � hide_dirs)r9 s r �
get_hide_dirsrE P s � � ��~�~�'�'�'r z� The standalone mode is not on par with the stable-diffusion-webui mode.14 Some features may be missing or work differently. Please report bugs.15c �( � dd l }|j |d�� d|vst |d � dk( rd|d< n d|d � �|d< t d|d � }|j j ||� dt
||� � |d � d |� �� }| j ||�16� y ) Nr T)�exist_ok�suffix� �-�samples_format� extension�.)�format)�os�makedirsr1 r r2 �joinr �save)�imager2 �basename�kwargsrO rN �fullfns r r r [ s� � ��17� � �D�4� �(� �6� !�S���)9�%:�a�%?� !�F�8� �!"�6�(�#3�"4� 5�F�8� ��)�6�+�+>�?����� � � �X�J�a� 8��x� H� I�&�QY�JZ�I[�[\�]c�\d�e�g��
�18�19�6�&�20�)r c �21 � dd l } | j j � rY| j j d� 5 | j j � | j j � d d d � y y # 1 sw Y y xY w)Nr �cuda)�torchrX � is_availablerA � empty_cache� ipc_collect)rY s r r r g sf � �� �:�:� "� "� $����"�"�6�*�
)��22�23�&�&�(��24�25�&�&�(�
)�
)� %�
)�
)�s �5A9�9Bc �D � t dz
a t t � t d��� S )Nr �04)�backbone_current_seq_numberr) � launched_at)�outpathrT s r r r s s% � �#�q�(�#��k�]�#>�r�"B�C�D�Dr c � � | S �Nr
)�fs r r r x s � ��r c � � dd l }t |j | � � D �cg c]0 }|j d� r�|j j | |� ��2 }}|D �cg c] }|j j
|� s� |��" c}S c c}w c c}w )Nr rM )rO �sorted�listdir�26startswithr2 rQ �isfile)�dirnamerO �x� filenames�files r r r z sm � ��7=�b�j�j��>Q�7R�l�!�Z[�Zf�Zf�gj�Zk�R�W�W�\�\�'�1�-�l� �l�!*�C��b�g�g�n�n�T�.B��C�C�� m��Cs �B� B� B �=B c � � |S rc r
�r r s r r r s � �w�r c � � |S rc r
ro s r r r � s � �7�Nr c � � dddddd�S )Ni@ �autocastF� � r" r
r
r r r- r- � s � � "� '� �&'�"$� &� &r c �@ � t t j dd� � S )NrM �outputs)�str�pathlib�Pathr
r r r3 r3 � s � �c�'�,�,�s�I�">� ?�?r c � � y rc r
r
r r r? r? � � � r c � � y rc r
r
r r rB rB � r{ r c � � i S rc r
r
r r rE rE � s � ��r ) rx r �enum�sys�Enumr �modules.imagesr �modules.devicesr r �modules.call_queuer r r r r r- r3 r? rB rE r �
USED_BACKBONE�printr) �now� timestampr` r_ r r
r r �<module>r� s� �� � � � 27��4�9�9� �28E,�)�(�7�7�(�� �
� �B�E�(� !�&�&�M��?,� � V�W�29*�)� �l�h�l�l�n�.�.�0�1�K�"#� �E�30 *�D�31 /� 3�&� @� � �"� !�+�+�M�s �?A) �)A!C