fred-dev/comfy_ui_ali
0
1�
2 ���g� � �^ � d dl Z d dlZd dlZd dlZd dlZd dlZdd�Zd� Z d� Z3d� Z d d�Z d4d�Z
y) � Nc �0 � t j |� }|�<t j | j � | j | j5 |d�� S t
j |d�� \ }}g }t |d dz � D ]d }t j dgt | j � � dd z | j | j6 |d�� }||v s�T|j |� �f |D �cg c] }|| �� }}t j |d� � }|S c c}w )7z�8 creates random noise given a latent image and a seed.9 optional arg skip can be used to skip and discard x number of noise generations for a given seed10 N�cpu)�dtype�layout� generator�deviceT)�return_inverse������ r )�axis) �torch� manual_seed�randn�sizer r �np�unique�range�list�append�cat) � latent_image�seed�11noise_indsr � unique_inds�inverse�noises�i�noises �W/Users/fredrodrigues/Library/CloudStorage/Dropbox/Code/comfyui_hardware/comfy/sample.py�
prepare_noiser s � �12 �!�!�$�'�I����{�{�<�,�,�.�l�6H�6H�Q]�Qd�Qd�py� CH� I� I� �9�9�Z��E��K��
�F�
�;�r�?�1�$�
%��� � �Q�C�$�|�'8�'8�':�";�A�B�"?� ?�|�GY�GY�bn�bu�bu� BK� TY� Z�� � � � �M�M�%� � &� ")�
)��A�f�Q�i��F�
)�
�Y�Y�v�A�
&�F� �M�� *s �, Dc �F � | j d� }|j |j d k7 rDt j |� dk( r,t13 j j ||j d�� }|j dk( r |j dk( r|j d� }|S )N�
latent_formatr r )�dim� � � ) �get_model_object�latent_channels�shaper
�
count_nonzero�comfy�utils�repeat_to_batch_size�latent_dimensions�ndim� unsqueeze)�modelr r" s r �fix_empty_latent_channelsr2 s� � ��*�*�?�;�M��$�$� �(:�(:�1�(=�=�%�BU�BU�Vb�Bc�gh�Bh� �{�{�7�7� �m�Fc�Fc�ij�7�k� ��&�&�!�+� �0A�0A�Q�0F�#�-�-�a�0� � �� c �: � t j d� | |||g fS )NzLWarning: comfy.sample.prepare_sampling isn't used anymore and can be removed��logging�warning)r1 � noise_shape�positive�negative�14noise_masks r �prepare_samplingr< # s � � �O�O�b�c� �(�H�j�"� 4�4r3 c �. � t j d� y )NzUWarning: comfy.sample.cleanup_additional_models isn't used anymore and can be removedr5 )�modelss r �cleanup_additional_modelsr? ' s � � �O�O�k�lr3 c � � t j j | || j ||| | j �� }|j |||||| | |
|||||��
}|j
t j j � � }|S )N)�stepsr �sampler� scheduler�denoise�
model_options)15�cfgr �16start_step� last_step�force_full_denoise� denoise_mask�sigmas�callback� disable_pbarr ) r+ �samplers�KSampler� load_devicerE �sample�to�model_management�intermediate_device)r1 r rA rF � sampler_namerC r9 r: r rD �
disable_noiserG rH rI r; rK rL rM r rB �sampless r rQ rQ * s� � ��n�n�%�%�e�5��AR�AR�\h�t}� HO� _d� _r� _r�%� s�G��n�n�U�H�h�C�l�gq� ~G� \n� }G� PV� ai� xD� KO�n� P�G��j�j��/�/�C�C�E�F�G� �Nr3 c �� � t j j | ||||| j ||| j ||| |17| �� } | j t j j � � } | S )N)rE r rJ rL rM r )r+ rN rQ rP rE rR rS rT )
r1 r rF rB rK r9 r: r r; rL rM r rW s
r �
sample_customrY 1 s� � ��n�n�#�#�E�5�(�H�c�5�K\�K\�^e�gm� ~C� ~Q� ~Q� `l� {E� PX� gs� z~�#� �G��j�j��/�/�C�C�E�F�G� �Nr3 )N)18g �?FNNFNNNFN)NNFN)r
�comfy.model_managementr+ �comfy.samplers� comfy.utils�numpyr r6 r r2 r< r? rQ rY � r3 r �<module>r_ s6 �� � � � � � ��&�5�m��r3