hololens/stable-diffusion-webui-depthmap-script
1
1o
2 ��f� � @ s@ d dl Z d dlZd dlZd dlmZ dd� Zdd� Zdd� ZdS ) � N)� OrderedDictc C s~ | dkrdS z+| � d�}t|�dkrt� |d W S dd�|dd� � }t�|�}t||d �W S ty> td| � � w ) z�Helper to return a function object by name. func_name must identify a3 function in this module or the path to a function relative to the base4 'modeling' module.5 � N�.� r zlib.�����z Failed to f1ind function: %s) �split�len�globals�join� importlib�
import_module�getattr� Exception�print)� func_name�parts� module_name�module� r �EF:\3dpicmaker\stable-diffusion-webui-depthmap-script\lib\net_tools.py�get_func s 6 7 8�r c C s� t j�| j�rHtd| j � t�| j�}|dur$|jt|d d�dd� |dur4|jt|d d�dd� |jt|d d�dd� ~tj �9� dS dS ) 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 �12checkpointr r r r s" ����r c sR t | �� �}t� fdd�|D ��s| S t� }| �� D ] \}}|||�� d�< q|S )Nc 3 s � | ]}|� � �V qd S )N)�13startswith)�.0�key��prefixr r � <genexpr>0 s � z*strip_prefix_if_present.<locals>.<genexpr>r )�sorted�keys�allr �items�replace)�14state_dictr+ r. Zstripped_state_dictr) �valuer r* r r" . s r" )r r r � collectionsr r r r" r r r r �<module> s