o
    i                     @   sd   d dl mZ d dlZd dlmZ d dlmZmZ G dd dej	Z
G dd deZdefd	d
ZdS )    )overrideN)ComfyExtensionioc                   @   s,   e Zd Zedd ZedejfddZdS )LotusConditioningc                 C   s   t jddg t jjddgdS )Nr   zconditioning/lotusconditioning)display_name)node_idcategoryinputsoutputs)r   SchemaConditioningOutput)cls r   6/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_lotus.pydefine_schema	   s   zLotusConditioning.define_schemareturnc                 C   s:   t  }tg dg dgg|}|i gg}t|S )N(   g     Կg     ܿg     g     0Ϳg     ?g     Ϳg     ̿g     @g     Կg     (?g          ؿg     g     ~?g     4ҿg      ?g     g     ˿g     ?g     `?g     I?g     g     |ؿg     ǿg      ¿g     g     x     xÿg     ʿ     οg     4Ͽg     ̻g     g     Ķ?g     οg     Ŀ     dۿg     g     ѿg     ƿg     ؿg      ?g     g     @g     `U     $?g     пg     \?g          ɿg     ѿg     D     Hݿg     Ϳg     r?     ׿g     |׿     Hʿg     Dÿg     ܿg     \ʿg     D?g     ӿg     Hÿg     ?g     g     xg     ԩ?g     Hg     ӿg     ǿg     pÿg     <?g     @?     g     g     ,g     ؿg     $g     ܺg     ǿg     ۿg      տ     g     P?g     8?g     `g     g     ҿg          пr         ƿg     hg     ÿ     ǿg     οg     ?g     ̿g     ȿg     ȿ     ۿg     xg     `޿g     пg     g     ܿg     ?     ǿg     տg     ӿg     ӿg     xƿg     ׿g          пg     ؿg          tԿg     g     ۿ     ǿg     ,ٿg     bg     ļ?r$   g     ̿     ȿ     \g     x?g     g     ѿg     ӿ     \пg     ?g     \     \ſg     ѿg     Hg     g     HϿg     ڿg     @Ͽg     Xg     ¿g     g     l¿g     g     g     ÿ     Ϳr   g     g     $     ƿ     Կ     ҿg     ിg     dڿ     ӿg     g     t?g     ̿     Xÿg     Dݿg     Hg     Ŀg     g      ?g     l?g     |g          0ӿg     ޿g     Tg     g     Pӿg     ֿg     пg     ܿg     $ٿg     ,r1   g     ?g     dѿ     (׿g     `п     Tʿg     dg     ?r   g     Ϳg     <ƿg     ٿg     DͿg     ,g     g     Lȿg          g     ſg     ޿g     g     ڿg     ?g     hi?g     {?     ʿg     ߿g     `@g     ݿg     ,ҿg     Tӿg     пg     ʿ     ÿg     \g      g     <?g     0ѿg     |?     Lѿg     pg     Lg          ݿg     ?g     t?g     0ſg     ڿg     ÿg     `g     <ӿg     @g     ߿g     <̿     ؿg     ~?g     ug     g     ¿g     ̿g     |ۿg     ٿr'   g     g     \g     tg     g     ?r5   g     ֿg     ſg     ҿg     ҿr   g     ڿg     ޿g     ʿg     ׿g     ǿg     `ۿ     |ƿg     4?g     tϿg     袿g     `p?g     g     r+   g     ̿g     Hɿg     Ŀr*   g     ؿg     g     4?g     ׿g     䒿     8ÿr(        hݿ     ÿg     ?g     D?g     g          Lҿg     xܿg     dοg     ̿g     ۿg     pg     ?g     L?g     dg     P          ѿr    g     ſg     pܿg     ˿g     Dѿ     ؿ     8Կ     ۿg     ѿr)   g     ſg     ?g     ȿg     {g     0οg     g     ̿r,        ޿g     տg     ?g     g     tg     ?g     $kg     8g     ,?g      ȿ     п     ؿg     g     g     0ؿg     Ͽg     ڿ     ?     տg     ?     ,ֿg     tg     ɿ     ؿg     ؜?g     t?g     g     ޿g     g     r%   g     Կg     ߿g     g     ƿg     ٿg     g     ?g     rB   g     g     ѿg     g     țg     hg     ?g     ؿg      rC   g     ڿg     Lؿg     Tɿg     g     xտg     ȱ?g     Կ     ȿg     X?g     ?g     Dп     @ӿg     g     g     ?g     4?     g     ӿg     ?g     ¿     Ŀg     <ʿg     g     Ͽg     |ÿ     пg     Ի?     \ӿg     lοg     ׿g     `ܿg     g     \g          Dտ     dƿg     ӿg     4ɿr   g     ܿ     ǿg     L|g     ,ʿg     Pпr   g      g     @g     ѿr-   g     L?g     Ԣ?g     ¿g     (ֿg     Ϳg     Ьg     ܿg      ̿g     赿g     ѿg     ÿg     ̿g     пg     (?g     $o     տg     g     8пg     Hg     g     ֿg     L?g     ˿g     Ϳg     ſg     ?g     ?g     8ɿr   g     g     ?r3   g     g     g      ?g     g     Ļg     |?g     ໿rQ   g     g     g     *@g     h?g     ,     ǿg     pg     h¿g     ƿg     Կ     ֿg     Ŀg     Կg     ?g          d׿g     lӿg     տg     Lڿg     hg     <     ҿg     ?     ˿g     ܘ?g     ̿rV   g     ƿ     ſ     Ͽg     g     g     տg     ˿g     0g     pg     0ٿ     g     Lʿg     xg      ǿg     \ٿg     <ɿg     g     g     ȣg     (g     g     ÿrJ   r%   g     l?g     <ٿg     x?     g      Ŀg     p?g     tg     οg     ҿg     l     Tҿg     @?r   g     Կg     ?g           g     P?g     |пg     Ͽr/   g      g     tֿg     H׿g     g     lͿg     ӿg     g     g     ٿg     ſg     $?g     Կg     dr]   g     tƿg     ɿg     8ƿg     8ʿrK   g     ,g     g     ءg     ɿg     ˿g     Ͽg     zg     ĸ     ݿg     ٿg     L̿g     g     (g     g     ¿g     ˿g     ҿg     ҿg     ƿrZ   g      ׿rY   g     lr        ҿg     `     ÿg     տg     g     ӿr(   g     Ⱥg     dֿg     ܻg     pſ     ѿg     Pg     8     ӿg     ƿg     пg     $?g     l̿r   g     lrR   g     Xοg     ޿g     οg     lg     Ͽg     ؿg     `ƿg     ׿g     Ѕg     ¿g     m     Կg     (ӿg     g     ҿg     $g     g     Xg     Xg     \ƿ      ҿ     Ŀg     0ܿg     ?r&   g     g     Ͽg     g     йg     Ŀg     ӿg     `ٿg     4?g     zre        Dֿg     g     g     ֿg     䇿rI   g     Pҿg     ǿr@   g     g     Lٿg     L     r!   g     ˿g     4ǿg     `?     ҿg     ?g     |¿g     LrS   g     Ȃ      ſrU   g     ?g     (?g     g     g     Dӿg     ܿg     pg     @?rb   g     ?g     Pοg     ÿg     g     hͿg     ҿg     ֿg     Dҿr:   g     dg     ڿg     8?g     ڿg     D?g     Ͽg      ?g     4g     g     g     g     g     ɿg     ׿r>   g     ƿr.   r   g     lg     ֿrA   g     İ?g     `?g     ʿg     t~g     ?g     pg     ˿g     g     Dοg     ſg     g     r2   g     Tg     0Կg     ѿg     hпg     g     $g     H޿g     g     g          Կg     r^   g     |?g     ſg     ؿ     P?     տrP   g     οg     ¿     PĿg     Ͽg     0g          Կg     T?g     lҿg     hrd   g      g     οg     $ҿg     lۿg     g     ſg     T?g     ?rA   r   g     g     g     ߿g     `?g     0ɿg     r0   g     ֿg     DԿg     g     ?g     ?g     g     g     Lg     g     \ؿg     տ     g     (?g     (?ro   g     hg     X?g     g     Ϳg     ǿg     L˿g     ۿ     ۿg     ƿg     $Ig     $ۿg     4rh   g     ¿g     ӿg     X׿     ׿g     ݿg     ܿg     dȿri        xۿg     x     p׿g     ۿg     r1   r:   g     ٿg          $пg     Dg     q?g     g     `ѿr#   g     tȿg     D?g     ?g     (g     տg     տg     p?g     0ȿg     ƿrt   g     ѿ     ڿg     ÿg      ѿg     xɿg     ƿg     ?g     ʿ     $ѿg     DrW   rT        ӿg     |Կg     g     ֿg     Crg   g     ǿg     ɿ     8ǿr   g     ؿg     g     $?g     P˿g     ӿg     ƿg     pӿg     r=        ҿg     \?g     ׿g     g     տg     `ʿg     䕿g     Dٿg     ̿g     g     xg     (ۿg     8g     οg     亿g     ?g     ؼ?g     D¿g     g     prF   g     ѿg     ܢg     Lg     ?g     0пg     g     ׿g     Կg     \?g     4     տrn         ֿg     g     ؿg     (?g     0?g      ʿg     Hٿr9        g     ?g     пg     <g     <g     g     ֿg     P     ƿg     @?g     ɿr&   g     ׿g     ?r}   g     Xҿg     ౿rP   g     ѿrk   g     g     g     <g     ֿg     ¿     (ݿrs   g     ѿg     ̿g     <ȿg     ?g     Ͽ     Xɿg     ֿg     Hӿg     8Ϳg      Ͽ     8g     (?      ӿg     rX   g     (g     ˿rx   g     g     ?g     $¿g     ?g     ?g     踿g     Lr"   r   g     Xg     PͿg      rM   g     ߿     xʿg     xg     ?r8   g     `Կg     Dۿ     ׿g     ˿g     ?g     `g     d?g     hT?g     t?g     g     Ϳg     ҿg     ?g     Ŀg     pg     пg     ÿg     Ŀg     dr+   g     g     ƿg     x̿g     8g     |g          <пg     տg     g     tg     @ʿg     ׿g     е?rv   g     l׿g     @ѿg     ?g     4g     8?r/   g     ?g     ſg     οg     ؿg     4˿g     ?g     L߿g     g     g     Կg     Pg     g     g     ؿg     ?g     8ٿg     g     r   g     hҿg     ׿g     Xg     ,̿r   g     \     H˿g     ?g     ̪r        ?g     Ͽr;   g     lg     տg     D?g     r{   (   g     ?g     ?g     tۿg     ?g     T?g     g     g     ?g      @     $g     <?g     g     g     g     g     ?g     Pg     H     g     ?rw   g     g     hӿg     ?g     ?g     4¿g     8@g     g     8?g          `g      g     ?r   g     xg           ?g     ?g     ?g     ?g     Lg     g     ׿g     $?g     ?g     g     ?g     ?r   r    g     g     Ŀg     (?g     g     Dʿg      ?g     ?g     <	@g     g     H @g     ?g      ?g     g     xe?g     ?r|   g     8?g     g     ?g     0?g     ?g     g     g     \?g     g     g     x?g           g     ?g     Կg     @g     ¿     g     l?g     g     Ըr[   g     ?g     g     g     <g     g     ?r   g     ҿr_   g     g     ?     g     hg     Ŀg     |?     ?g     g     ?g     ?g     8?g     ?g     Xg     0ڿg     4@g     `ӿg     tg     ?g     ?g     ?g     g     Cg     ?g     T @     ?g     g     ?g     X?g     p@g     g     g     xg     Hg      g     $?g     x޿g     g     L?     g     dg     g     \g     0g     g     $g     H     g     g     g           g     g     ܯ?g     x?g     Tڿr   g     xg     Xrz   g     8޿g     4g     Dg     dg     ?g     pg     ?g     T?g     ?g     g     xg     <?g     P?     g     @g      g     ?g     Hҿ     g     g     @     g     ?g     ?g     tg     ͿrL   g     g     g     |?g     ?g     `r@   g     Dr   g     g     ?g     пg     g     ?     hg     ,?g      ?g     g     ?g     |g     g     tg     r   g      g     ?g     g      g     dg     ?g     p?g     ?g     ?     g     \?     g      ߿g     @g     ?g     ¿g     r   g     g     ?g     g     пg     @g     t?g     ?g     g     ?g     pο     4?g     ?g     4?g     Tg     xg     ɿg     P?g     ÿg           ?g     L?g     dſg     g     g     ?g     L¿g     g     ?g     g     pg     X?g     x?g     g          ?g     ۿg     ?g     ?g     l?     pg      ?g     g     Ϳg     ?g     ?     ?     g     ?g     lg     ܿg     g          ?     ?g     @g     0g     g     g     ܿg     ҿg     @g     g     ?g     Xg     g     @g     Hg     g     ?g     g     g     H?     d?r   g     \ @rr   g     <?r   g     `g     x?g     g     g     ?g     \?g     ?g     ̷r        g     4g     ?g     ?rc   g     g     g     g          hڿrX   g     r   g     ?g     ?g     ?g     pg          pg     	g     T?     0g     (g     d?g     g     @?g     H?g     $?g     x?g     ?g     ?g     ?g     @?r   g     lg     g     \?g     ѿg     \?     g     g     ?g     ?g     $?r8   g     (g     g     g     ?g     ?g     g     Կg     g     g     g     rH   g     `g     pȿg     ,@g     r   g     \ g     ۿ          @ſr`   g     Xg     @g     |g     xr   g     hg     r   g     ?g          g     g     ?g     ,g     r   g     @?r        8g     (g     ?g     ؿg     pg     P?     dg     ?g     g     @r6   g     `?g     ?g     @g     d?g     ?g     ?g          Tg     ?     ?g     ۿg     xg     Թ?g     ?g     xg     ?g     ?g     ۿg     p     Xg     P?g     g     g     g     ?g     g     g     g     Hg     ޿g     ,?g     d?g     (g     d?g     ?g     pg     g     ?rp   g     dg     ?g     8?g     g     8?g     <g     g     P?rO   g     ׿r   g     g     d?g     Pɿg     g     g     `?     ?g     ?g     ȷ?g     8?g          ?g     \ҿg     ?g     ٿg     g     g     ?g     @ҿg     ׿g     ?g     ?g     (i?g     ?g     t?g     g     ?g     r   g     d?     g     g     8g     g     ?g     ?g     r   g     ?g     g     0?g     g     ?g     g     g     ۿg      ?r   g     $g     ?g     s?g     X?g     g          ?g     ?g     D?g     ?g     g     ra   r   g     |?g     ѿg     ?g     @g     tg     g     ?     g     l?g     g     ֿg     g     g     Hܿr   g     Pg     dg     P?g     g     ?g     8r   g     g     Ͽ     ?g     ?g     d
@     ?g     g     hg     ?g     g          g     8?g     L?r-   g      ?g     ?rN   g     ÿg     |r   g     D?g     D?g     g     r   g     ?g     ?g     r~   r   g     ?g     Lӿg     g     h?g     g     ?g     ?g     ٿg     ?g     g     pg     ʿg     g     ,?g     ?g     ̿g     ?g     ܿg     ?g     g     Pg      g     4߿g     g     ?g     g     ÿg     ?     d?g     ?g     ?g     ?g     L?g     ?g     <?rl        g     g     g     (g     ?g     ?g     ?g     tg     ,?g     HԿg     @?r   g     ?g     ?     ,ݿg      g     g     (?g     g     0g     ?g     ?g     ?g     ?g     ,r   r   g     r   g     ?     g     g     D޿r7   g     ?g     g     ?r<   g     8ݿg     g      g     g     ?g     ?     r         g     $?g     Xg     (?g     l?g     ?g      @g     ,g     g     ?g     ƿg     ?g     LԿr   g     ?g      g     ,?g     $?g     ?g     ?g     h?     r   g     ?g     ?g     r   r   g     L?g     տg     r?   g     r   g     |ٿry   g     g     ?g     ?r   g     ܕ?g     (?g     g     ?g     g     ?g     Ͽg     ?g     P?g     0g     g     ?g     $g     ҿg     ?rj   g     ?r   g     ?g     p?g     r   g     X?r   g     g     Pr   g     r   g     g     r\        ,ӿg     ۿg     (޿g     ?g     ?g     `?g     T?g     ?g     tg     g     ?g     Tٿg      g     ſg     pg     @?g     lr   g     @g     `?g     ?g     d?g     ?g     ?g     (g     ?g     g     g     ?g     g     l޿r^   g     pg     h?g     t߿g     g     tg      g     Lÿg     ?g     hrm   g     g     4ٿg      g     @g     p?g     t?g     g     ?g     ؿru   g     ?g     ?g     Dg     ?g     g     g     $?g     g     4?g     ?g     g     Pg      @g     r4   g     g      ?r   g     g     ?g     <g      ?g     ?g     g      ?g     x?g     ?g     dg     0g     ʿg     ?g     Hrf   r   g     |?g     g     g     $@g     ?r   g     `?g     ڿg     (g     ѿg     g     g     ?g     Hr   g     ?g     ?g     g     r   g     @?g     ?g     g     g      ÿg     g     g     ?g     訿g     g     跿g     ?g     h?g     8?g     ?r   g     h?g     ?g     ?r   r   g     L?r   g     ֿr   g     ?g     ?g     ѿg     蔿g     g     ?g     ?g     \?g     Pÿg     (տg     8?g     ?g      ?rE   g     ?g     x?g     g     X?g     ?g     ?g      g     ?g     r   g     <?g     (g     rV   rD   r   r        ?g     ?g     p?r   g     ?g     g     Dڿg     ɿg     4ſg     в?g     ?g     g     4g     r   g     pg     Lg     g     8ۿg     r   g     ?g     Pܿg     g     ?g     \r   g     r   r   g     g     hr   g     $g     @g     ֿg     Hѿg     ( g     Lg     ?g     ѿr   g          ?g     ?g     dr   g     \?g     `?g     ?g     g     ?g     ?g      ?g     $?g     ?g     xg     g     Tr   g     dg     ?rq   g     |g     ?g     @g     g     g     p?g     g     rY   r   g     ?g     ̿g     H?r   g     g     Tg     Lg     H?r   g     ߿g     Tÿg     ?g     0ۿg     $߿g     ?g     d?g     ̽g     d?g     ?g     칿r   g     ?g     g     ?g     @ݿg     ?g     g     hr   g     ?g     ?r   g     $̿g     ?g     ?g     `r   r   rG   r   g     ?g     g      g     g     r   g     <տg     \?g     T?g     ƿg     8?g     hg     ?g     ,?g     0ֿr   g     L?g     ?g     0?g     <g     ?g     g     ?)mmget_torch_devicetorchtensortor   
NodeOutput)r   deviceprompt_embedscondr   r   r   execute   s   

zLotusConditioning.executeN)__name__
__module____qualname__classmethodr   r   r   r   r   r   r   r   r      s
    
r   c                   @   s(   e Zd Zedeeej  fddZdS )LotusExtensionr   c                    s   t gS N)r   )selfr   r   r   get_node_list   s   zLotusExtension.get_node_listN)	r   r   r   r   listtyper   	ComfyNoder   r   r   r   r   r      s    r   r   c                      s   t  S r   )r   r   r   r   r   comfy_entrypoint&   s   r   )typing_extensionsr   r   comfy.model_managementmodel_managementr   comfy_api.latestr   r   r   r   r   r   r   r   r   r   <module>   s    