hololens/stable-diffusion-webui-depthmap-script
1
1o
2 ��f� � @ sF d dl Z d dlmZ d dlZd dlZG dd� dej�Zz@d dlmZ d dlm Z d dlm3Z4 d dl m Z d d l
mZ d5d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZejZW dS ed� dd� Zdd � Z ee�� �� �Z d a d d � Z6d d � Z d!d"� Zd#d � Zd$d
� Zd%d� Zd&d� Zd'd� Zd(d� Zd)d� Zej ZY dS )*� N)�datetimec @ s e Zd ZdZdZdS )� BackboneType� � N)�__name__�7__module__� __qualname__�WEBUI�8STANDALONE� r r �DF:\3dpicmaker\stable-diffusion-webui-depthmap-script\src\backbone.pyr 9 s r )�10save_image)�torch_gc)�get_next_sequence_number)�wrap_gradio_gpu_call)� listfilesc C �$ ddl m} t|| �r|�| �S |S )Nr )�opts)�modules.sharedr �hasattr� __getattr__)�name�defaultr r r r �get_opt s 1112r c C r )zGet command line argumentr )�cmd_opts)r r r �__getattribute__)r r r r r r � get_cmd_opt s 1314r c C sd i } dD ]}t d| d�}|du rt|d�}|dur || |< qdD ]}|| v r/t| | �| |< q!| S )z"Parameters for depthmap generation�Z15boost_rmax� precisionZno_half�marigold_ensembles�marigold_stepsZdepthmap_script_N)r r )r r �int)�ops�s�cr r r �16gather_ops$ s 17��r% c C sD t dd�} | du st| �dkrt dd�} | dur t| �dks J �| S )z+Get path where results are saved by defaultZoutdir_samplesNr �outdir_extras_samples)r �len)�pathr r r � get_outpath4 s18 1920r) c C s^ ddl m} m} | jd ur+| jjd ur| jj�|j� | jjd ur-| jj�|j� d S d S d S �Nr ��shared�devices)�modulesr, r- �sd_model�cond_stage_model�to�cpu�first_stage_modelr+ r r r �unload_sd_model= s 21 �r4 c C sZ ddl m} m} | jd ur)| jjd ur| jj�|j� | jjr+| jj�|j� d S d S d S r* )r. r, r- r/ r0 r1 �devicer3 r+ r r r �reload_sd_modelG s 22 �r6 c C s dd l } | jjS )Nr )r r, Z hide_dirs)r. r r r �
get_hide_dirsP s r7 z� The standalone mode is not on par with the stable-diffusion-webui mode.23 Some features may be missing or work differently. Please report bugs.24c K s� dd l }|j|dd� d|vst|d �dkr d|d< n d|d � �|d< td|d �}|j�||� dt||�� |d � d |� ��}| j||d25� d S ) Nr T)�exist_ok�suffix� �-�samples_format� extension�.)�format)�os�makedirsr' r r( �joinr �save)�imager( �basename�kwargsr@ r? Zfullfnr r r r
[ s 26"�r
c C s\ dd l } | j�� r,| j�d�� | j�� | j�� W d � d S 1 s%w Y d S d S )Nr �cuda)�torchrG � is_availabler5 Z empty_cacheZ ipc_collect)rH r r r r g s 2728 "��r c C s t d7 a tt� t d���S )Nr Z04)�backbone_current_seq_numberr! � launched_at)�outpathrE r r r r s s r c C s | S �Nr )�fr r r r x � r c s8 dd l �� �fdd�t��� ��D �}�fdd�|D �S )Nr c s$ g | ]}|� d �s�j�� |��qS )r> )�29startswithr( rB )�.0�x��dirnamer@ r r �30<listcomp>| s $ z listfiles.<locals>.<listcomp>c s g | ]31}� j �|�r|�qS r )r( �isfile)rQ �file)r@ r r rU } s )r@ �sorted�listdir)rT � filenamesr rS r r z s r c C � |S rM r �r r r r r r rO c C r[ rM r r\ r r r r � rO c C s dddddd�S )Ni@ ZautocastF� � r r r r r r r% � s �c C s t t�dd��S )Nr> �outputs)�str�pathlib�Pathr r r r r) � s c C � d S rM r r r r r r4 � rO c C rc rM r r r r r r6 � rO c C s i S rM r r r r r r7 � rO ) ra r �enum�sys�Enumr Zmodules.imagesr
Zmodules.devicesr r Zmodules.call_queuer r r r r r% r) r4 r6 r7 r Z
USED_BACKBONE�printr! �now� timestamprK rJ r32 r r r r �<module> sL 33 �