hololens/stable-diffusion-webui-depthmap-script
1
1�
2 ��f� � �: � d dl Z d dlZd dlZd dlmZ d� Zd� Zd� Zy)� N)� OrderedDictc � � | dk( ry | j d� }t |� dk( rt � |d S ddj |dd � z }t j3 |� }t
||d � S # t $ r t d| � � w xY w) z�Helper to return a function object by name. func_name must identify a4 function in this module or the path to a function relative to the base5 'modeling' module.6 � N�.� r zlib.�����z Failed to f1ind function: %s) �split�len�globals�join� importlib�
import_module�getattr� Exception�print)� func_name�parts� module_name�modules �EF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\net_tools.py�get_funcr s� � �7 �B��� �����$�� �u�:��?��9�U�1�X�&� &� �s�x�x��c�r�8� 3�3� ��(�(��5���v�u�R�y�)�)�� � �
�,�i�8�
��s �.A2 �:A2 �2B c �� � t j j | j � r�t d| j z � t j | j � }|� |j t |d d� d�� |� |j t |d d� d�� |j t |d d� d�� ~t9 j j � yy) z10 Load checkpoint.11 zloading checkpoint %sN� shift_modelzmodule.T)�strict� focal_model� depth_model) �os�path�isfile� load_ckptr �torch�load�load_state_dict�strip_prefix_if_present�cuda� empty_cache)�argsr r r �12checkpoints r r r s� � � 13�w�w�~�~�d�n�n�%�
�%����6�7��Z�Z����/�14� � "� � '� '�(?�15�=�@Y�[d�(e�+/�
(�
1� � "� � '� '�(?�16�=�@Y�[d�(e�+/�
(�
1��#�#�$;�J�}�<U�W`�$a�+/� $� 1� �
�17�18� � � � &� c �� �� t | j � � }t �fd�|D � � s| S t � }| j � D ] \ }}|||j �d� < � |S )Nc 3 �@ �K � | ] }|j �� �� � y �w)N)�19startswith)�.0�key�prefixs �r � <genexpr>z*strip_prefix_if_present.<locals>.<genexpr>0 s �� �� �6�#�s�~�~�f�%�6�s � r )�sorted�keys�allr �items�replace)�20state_dictr/ r2 �stripped_state_dictr. �values ` r r$ r$ . sh �� � �*�/�/�#� $�D� �6��6� 6���%�-�� �&�&�(� =�21��U�7<��C�K�K��� 3�4�=� � r) )r
r! r � collectionsr r r r$ � r) r �<module>r; s �� � � � #��(!�& r)