fred-dev/comfy_ui_ali
0
1�
2 ���gr � �r � d dl Z d dlZd dlZ G d� d� Z G d� de� Z G d� de� Z G d� d e� Zy)3� Nc �* � e Zd Zd� Zd� Zd� Zd� Zd� Zy)� CONDRegularc � � || _ y �N��cond��selfr s �V/Users/fredrodrigues/Library/CloudStorage/Dropbox/Code/comfyui_hardware/comfy/conds.py�__init__zCONDRegular.__init__ � � ��� � c �$ � | j |� S r )� __class__r s r �4_copy_withzCONDRegular._copy_with5 s � ��~�~�d�#�#r c � � | j t j j | j |� j |� � S r )r �comfy�utils�repeat_to_batch_sizer �to�r6 �7batch_size�device�kwargss r � process_condzCONDRegular.process_cond
s2 � ����u�{�{� ?� ?�� � �:� V� Y� Y�Z`� a�b�br c �` � | j j |j j k7 ryy�NFT)r �shape�r8 �others r �9can_concatzCONDRegular.can_concat s" � � �9�9�?�?�e�j�j� .� .� .��r c � � | j g}|D ] }|j |j � � t j |� S r )r �append�torch�cat)r10 �others�conds�xs r �concatzCONDRegular.concat s6 � ���� ���A� �L�L���� � ��y�y�� � r N)�__name__�11__module__� __qualname__r r r r! r) � r r r r s � ��$�c��12 r r c � � e Zd Zd� Zy)�CONDNoiseShapec � � | j }|�=t |� dz }t |� D ]! }|j |dz |||z || � }�# | j t13 j j ||� j |� � S )N� ) r �len�range�narrowr r r r r )r14 r r �arear �data�dims�is r r zCONDNoiseShape.process_cond s| � ��y�y�� � ��t�9��>�D��4�[���{�{�1�q�5�$�q�4�x�.�$�q�'�B�� !� ���u�{�{� ?� ?��j� Q� T� T�U[� \�]�]r N)r* r+ r, r r- r r r/ r/ s � �^r r/ c � � e Zd Zd� Zd� Zy)�
CONDCrossAttnc � � | j j }|j j }||k7 rN|d |d k7 s |d |d k7 ryt j |d |d � }|t |d |d � z }|dkD ryy)Nr r1 F� � T)r r �math�lcm�min)r15 r �s1�s2�mult_min�diffs r r! zCONDCrossAttn.can_concat' s� � �
�Y�Y�_�_��
�Z�Z�
�
��
��8��!�u��1��~��A��"�Q�%� � ��x�x��1��r�!�u�-�H��s�2�a�5�"�Q�%� 0�0�D��a�x� �r c � � | j g}| j j d }|D ]B }|j }t j ||j d � }|j |� �D g }|D ]H }|j d |k r#|j d||j d z d� }|j |� �J t
j |� S )Nr< )r r r> r? r# �repeatr$ r% )r16 r& r' �crossattn_max_lenr( �c�outs r r) zCONDCrossAttn.concat4 s� � ���� �� �I�I�O�O�A� .���A����A� $���):�A�G�G�A�J� G� � �L�L��O� �17 ���A��w�w�q�z� -�-��H�H�Q� 1�Q�W�W�Q�Z� ?��C�� �J�J�q�M� � �y�y��~� r N)r* r+ r, r! r) r- r r r: r: & s � � �
r r: c �$ � e Zd Zd� Zd� Zd� Zd� Zy)� CONDConstantc � � || _ y r r r s r r zCONDConstant.__init__D r
r c �8 � | j | j � S r )r r r s r r zCONDConstant.process_condG s � ����t�y�y�)�)r c �8 � | j |j k7 ryyr r r s r r! zCONDConstant.can_concatJ s � � �9�9��18�19� "��r c � � | j S r r )r20 r& s r r) zCONDConstant.concatO s � ��y�y�r N)r* r+ r, r r r! r) r- r r rK rK C s � ��*��21r rK )r$ r> � comfy.utilsr r r/ r: rK r- r r �<module>rQ s> �� � � � � �*^�[� ^� �K� �:
�;�
r