From 7d307d5673b2d4fc2c5bb0ea904a26c42445ea1e Mon Sep 17 00:00:00 2001 From: alpertunga-bile Date: Tue, 5 Sep 2023 21:27:50 +0300 Subject: [PATCH] fixed clone and run bug --- __init__.py | 44 ------------------ __pycache__/__init__.cpython-310.pyc | Bin 0 -> 907 bytes __pycache__/__init__.cpython-311.pyc | Bin 2801 -> 0 bytes __pycache__/preprocess.cpython-310.pyc | Bin 0 -> 1763 bytes __pycache__/preprocess.cpython-311.pyc | Bin 3374 -> 0 bytes __pycache__/prompt_generator.cpython-310.pyc | Bin 0 -> 5453 bytes __pycache__/prompt_generator.cpython-311.pyc | Bin 9458 -> 0 bytes generator/__init__.py | 38 +++++++++++++++ .../__pycache__/__init__.cpython-310.pyc | Bin 0 -> 997 bytes .../__pycache__/generate.cpython-310.pyc | Bin 0 -> 2279 bytes .../__pycache__/generate.cpython-311.pyc | Bin 3370 -> 0 bytes generator/__pycache__/model.cpython-310.pyc | Bin 0 -> 1206 bytes generator/__pycache__/model.cpython-311.pyc | Bin 1895 -> 0 bytes generator/__pycache__/utility.cpython-310.pyc | Bin 0 -> 920 bytes generator/__pycache__/utility.cpython-311.pyc | Bin 1408 -> 0 bytes 15 files changed, 38 insertions(+), 44 deletions(-) create mode 100644 __pycache__/__init__.cpython-310.pyc delete mode 100644 __pycache__/__init__.cpython-311.pyc create mode 100644 __pycache__/preprocess.cpython-310.pyc delete mode 100644 __pycache__/preprocess.cpython-311.pyc create mode 100644 __pycache__/prompt_generator.cpython-310.pyc delete mode 100644 __pycache__/prompt_generator.cpython-311.pyc create mode 100644 generator/__init__.py create mode 100644 generator/__pycache__/__init__.cpython-310.pyc create mode 100644 generator/__pycache__/generate.cpython-310.pyc delete mode 100644 generator/__pycache__/generate.cpython-311.pyc create mode 100644 generator/__pycache__/model.cpython-310.pyc delete mode 100644 generator/__pycache__/model.cpython-311.pyc create mode 100644 generator/__pycache__/utility.cpython-310.pyc delete mode 100644 generator/__pycache__/utility.cpython-311.pyc diff --git a/__init__.py b/__init__.py index 9110b76..7306dcb 100644 --- a/__init__.py +++ b/__init__.py @@ -11,15 +11,6 @@ path.append(dirname(__file__)) from prompt_generator import PromptGenerator - -def check_package(package_name: str) -> None: - if find_spec(package_name) is None: - print(f"/_\ Installing {package_name}") - process = run( - f"pip install {package_name}", shell=True, check=True, capture_output=True - ) - - print("/_\ Loading Prompt Generator") # Check prompt_generators folder under models folder @@ -32,41 +23,6 @@ if exists(root) is False: if exists("generated_prompts") is False: mkdir("generated_prompts") -# Check required packages - -# Installing from git because there is a error with normal transformers package -if find_spec("transformers") is None: - print(f"/_\ Installing transformers") - process = run( - f"pip install git+https://github.com/huggingface/transformers", - shell=True, - check=True, - capture_output=True, - ) - -check_package("accelerate") -check_package("xformers") - -if os_name == "Linux": - check_package("triton") - -check_package("optimum") - -if find_spec("onnxruntime") is None: - print(f"/_\ Installing onnxruntime") - process = run( - f"pip install optimum[onnxruntime]", shell=True, check=True, capture_output=True - ) - -if find_spec("onnxruntime-gpu") is None: - print(f"/_\ Installing onnxruntime-gpu") - process = run( - f"pip install optimum[onnxruntime-gpu]", - shell=True, - check=True, - capture_output=True, - ) - # Import PromptGenerator node to ComfyUI NODE_CLASS_MAPPINGS = { diff --git a/__pycache__/__init__.cpython-310.pyc b/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..762a1ad245a00b1f23df90c32cbdd9ca6c7a46b6 GIT binary patch literal 907 zcmZWn&x;c=6rRkFcG{hG+iqQD@i5m4Tfw7^{o9Iz(egL@UOv52=|3pk{;bzqx- z&-=t9oh|sXt6OJX-h(|8KfvMEa3sJD+}ygbPkhqf0=!o^@`Z^;r2o;ut;)7P1(mj8 z6(+~W`&Y_Wq+zkyyF+>uz+1S3J7zt^UEEu_@HRe}JA?iYb3$hTNw#C+F_l@u zNm(QBsnV+S-w2u+fhP;Q1pKX07|=**hLOt2(l|A>R+?~w_GeP?T%VDQ$V3adxqBZi z;)zHw%~tU)7+*)r8W)%_1-p5bhE{*`)YFY`GdB-ns^X<2i6(d$&lpJ*i4%fMB#!s!iv=e>j*v8=s<+$!I=D7o*wie0nx7k1BF}KA%lS zFVS>#ae5t_4(^nmIk0q1>&m)NL)$xP9+D${YXZwRRdH$dToR?EeNUw;l&o9OraTu? zGkQ5JG-WchWSFx=r$Tbsu&$|!cGr;PImvJtB9u}_5Q3fENp4yt?PD&3T+&RJz7#^s z$S|^16+$Rs456=%JhG%Uzb4~{i@8oN85z?wEff{m;zaOt{rr5fER+@;W#(SgVz<@t zb$gGOR#ngiLNucqp<%u*!$;;43q~FTQ(N6LCUW{9y83Ev+l}1NZTb;71R*dO_<{f1 H&dC1*!6yOq literal 0 HcmV?d00001 diff --git a/__pycache__/__init__.cpython-311.pyc b/__pycache__/__init__.cpython-311.pyc deleted file mode 100644 index 525c667dc88323a9de90ae2874ed92d7db7ee3a6..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 2801 zcmahKO>f-Bb+{ZZ_gi19&z07UE!kRsXn~+D(AG#{#ExRvR)orpn{8cowKFTxCYQvJ z3rLF;*aALmfgW^AfFOWRZiD0yAea1r9(o)W1iBz#KtK*fZ<{)&oce}K+oEcx!#8i{ zy?O7=dvE3q|57aG5VU`hf0F;G2z?`n=MCtnJ^NbjH6%&QQ)bGN8?uEP z*isrw#FNdmr8d-Pj?IjfZDgalV&<%TBOlGvX2B{piV{j8T6_W(IuP@@Dq>A}Qv z22e7cd?JAi^l?j4kww*N%{U# zwEt|98_B9jtviJ&}GFdv!r6p2%7-fP9zsh9!v_!;Rc?CC5 zs!49_?U^Ju_E!4=Z}a4hq)u$xx0K9)bi+vTdV7bl0IRAkyxk z>ro446Wh$T9B-3$C~JB)3ky!O^{}}|^|4h(y)Nr)wO(9b*gg$ihW_jT&_~ESvGyTK z2HU^r3CXw-=|l7h`ZSf4k&vf6mq1XKV;Lfg?+cu~7JrZk+JWMQR47}$iMsw4u*q!>@ zw$(oT)y;aV>w31OcWgr4I@p2~U)g(}<0~y8)ivt6ZgdP!*EgKQuo$_h$Ju9du&;PV z+z$Zt(NlHuXxUe*fm-G2`KOs8FKs+700Qtc^&nH{nflY|*?Jo3!attx1S>Xm^=5R?}XpX`Ze1RIP2BgtBldUZ6xDOS)mf zW6h4+wwXnl+pC;9+llj`e%sE3+61K?=x570(VXN7JCy&@m9HC zK~;}rNs@-B%#$!gHU4@Sq9vY$S6Nu)C;>xM;Ym2oO%BfgY5mXZef1dS`}k3QusPT~ zde<+k1cenJRRdJzsCtYp^CXPYij;dSj}Q!h#nEsYO(H4XS9p5WM{5CE<7jP^L776| z8@$WsYQB0gP%rlJP|o&$GFbiN)kjy4e$3?sfFaKIZ}8mvM`b_&e~(9)`uJXe?{R!D zk^!|_k8bsE9Sa*gx8lpyK(0Psk`AUE*99NDbQNu?I0b1Z_;aDz!KlgpP637)UpNsvu z;N$fGuXDT}`*Z2bAARldyT67vRCogj-q6UKOcW9?R(<(QAfI^|_9Y)*4)A4;FUL0O tKM+$HDf1ssfc5zlM>CKoWu7lw1iA|J5*%CM4Sx15p8G{#6<>6z{|kSH_9_4X diff --git a/__pycache__/preprocess.cpython-310.pyc b/__pycache__/preprocess.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..c8c7063786e10551f9812b34b0aceacf28756540 GIT binary patch literal 1763 zcmZ`)-ESL35Z~Eb-<=&hDS@;!6cM#Z^dOB&OZie1ff9ibl3xNmSSglsyAE#7=gaQ7 zB;hO-r4lV~JR>CgF%qx*1N<5L$`j8OLW%@pX6+;nf<5hcc6MfWK7Kpy)oK-nU+9@=h;}xb;kb+Xiv3n{^Up zL!VZKge`%vPo`w*1zPN|kQqJ$(yfZ6EYI zxS?<_R>rJk2ez`BbWBcpSfjESa``$tys@M5WEJQJdN!d0(Bgt6~KO#bY;z#h5<-A}+ zPx28Q!2ovH0MoX??teJ&q&E;!427!+VI?d)s>uk3r4bvH3dGSa*U{d5!O>!A;1^!u zqvW0p28guu%9(ZOtoV7)Wg}pZlp*IUY#8XCyTRIi1U#{EA?Kkh-ee2%a z@4Wlo2`qnR*@f^ZxfI%6oX@Ty`Ihidp5OsAhVuo(zuc4VtgJs~@UgbiCO!-sOoC|OlRndHEU5ilV~9#Nr^I82 zov$uS3;&r7f}MFC$t=uz0z#bEc*r?~JOqIsuG}yGTZjtqM}q4csBY)ZA5*i7&n~8j zPlp3LF4l3H?~&vPWCSjO7~6*QMttBE9`e}U&Uk-H9?0-aULMQw32<2|ydw{7K~@}u z!=aZ8lp);d#7*oeat3$SCGDX0(8rVT?$iCKT0bn`T;4L#H+Pl?e~{O|4o_goLgkPz zFP%X71PV<+I^;`BwsrwLwmM#p>w>kWKy83+1!1W)0bDyMP>NjnHDNsxuFU#Xr=7L( zT*q8Uqcf^2n(Ugt8n^-d=`WV|9Szi9xJ$Ak1EzDY^6I6 wdzR~oKh8z$H@N9qkB5vap|hL1`uzTI6)W0L)Ec#aL;q-1b_wSBWwH3mKcH)zHvj+t literal 0 HcmV?d00001 diff --git a/__pycache__/preprocess.cpython-311.pyc b/__pycache__/preprocess.cpython-311.pyc deleted file mode 100644 index 87839779d0a2af0a1add3d41ab219cff2c1cc505..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 3374 zcmb_eOKcm*8J^i)?s7?*WGK?MEXbBBA~K@~9Y5L_m55q4%(za80CLmRsYOuSkwTlV z%q~+&Sr$A52yN(~8VIDcF~a8HFm-^aL4f#J^icFj3R{R+G=PBikmN)_2w>=xf0opP zk`lORf6V?fJCA?ne|<9_9XaAaK>t$Km6C|izuC_=!M{bD(@-rV1t~aC6|hNv)93Qn_r1GkLWzZA!^pCZ9?Z)2@;?3M88(rbvlapxGqe*#aRB+B|%l zW5C|Q%ShWZ`Vjga!PlaDxQu78g5S35=nCo_gVS)a4#759)CJ$>RbcOe;8~=jTfIfB zG>G&ZW;nB`9R(o{(dv5XMB>I@?$|n>h#GBmc857T$f=WnMWeW85 zmA0nCujeu(tfnYYrwI-JVtz88NRsfRmY7bHE2){8f(j-IU(RJ_=6`uDoGhqXE)&n@ z6rzUn6bz<~%#tjj2`xuQz<@LJg%k`B7)ugW9nH^gImXf{Rb$4DEDX#NEndi`z`k+T zJ35xmB@<~i0*&oSEGX1rU{6Y@=A3AirT!|e-l)n|dG*E@uYUIGLXobb!!4L1)JHtX_r)bk3A;=$aogTYq*O>R)q@xOx zbbyq7t2$*A@hmEGI=+H#{i(<+d{Iz@3RCU?mF*CeHV8_7e^7Qxtn4|{u8@b&+zs)f zO^39+W9dinVnpX)|7%52=XJ5&D=1)CHA^lcqcQdV9y1rzy$g-Xw7{mq=>yJ7oU1IjX385@>rS3u<))*psi&R|s;bI|`+Ot8YAZ*TPq! ztn|0)Hn+yDxW4+oX-7}Mu9kkZhaIAgGL|azB=kJJ61{BwUL}U6YC}`2{eMmT<9D^I zx1dxnwdyvv#w{g667{iju-&Z;H*IP@ozhI%gA({GNqA9h| zOf_Bk0wwY7Ke5$-%OR1C1CW|h+w$xR*_mes$ec#OFGPv&>86$zGKlYMQyYi^hx(EZ z!Av`R>W=_J7#szTKe%4&KG!PszTh_C)=q@iJmH!ryep__L*6q@J92e3 z5tdIkT*ud4r`BAjDq`I=Xt)MzuE8f8&SN#=>`yIMJ}9^2e&G}w zU>?}z5wnd&pb=IaZQLAC?1$e#8jIyN?MKZ{3@!56pMpeZTjGW9mFV1-hY;BGP<#f> z!7sb-n;-gPo&n@`#TMk#!@p5OZ=D@3e#`v!x5^go+ro|oZWta??O*fuSAtcc?j13_Bb}Xm-8*b}hxc|4RMf+k@67zt+qD#3xly^eEpJ0Qx)NTK zLNzJW@E==#@&1ed>^-}BvEDmw^p4m45yKxT%VoJCA76>B$dBf?i34SYcV;PAwPL3WwJ zqfVnOYcor=E*SOZu)8ngWL};}xxm wSX~ub^&#Bei;q_To1m)ttI1Egt>%6f@!-HxvWcLoPFmG!tFqdMG0P_30b*?vHUIzs diff --git a/__pycache__/prompt_generator.cpython-310.pyc b/__pycache__/prompt_generator.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..adc786835693072798cce63f7c0123b3faf1db8b GIT binary patch literal 5453 zcmds5&2JmW72la%lFQ|ilBjRXAGDpgW+E$JMbjFt9oMo|0f}Ttbej+k7BpwXti9Z& zXP34m41uEZp_ia(0~BbR08)V-`X}^1$f<`OdYA(3A#G8&hZZQ1q)u!1z1^imGfGh) zx6Wc8-~8U2+4*>H-WvwRqNPDuy7@6XazfKS#L8gNq4Em6;R+C@G2&@$QYQp?>e04d z*V{(jXlLq~7}LFMJ6q4jyy4~AX5DP(>v^IrX)MFC_cfN?rge+u*0rkn0Q%IbMCCj; zjF>B`x-xDCuCFp~2vwEk_dpPWhSd?gBZ4Lm!zxww68E`iMEs&?g({!c0@1|&21^+# zufQ9A2ME{dglTokbw(JyuWggM!Ss5D8TBm7)N?G$O_t+%W^#+=d4XBnW(DAFZf%rU zk(KTn^&%Sq1T{wC9fP;8Rfd-1P`BB{Mwv~rDbOylX*P3TtBit=}jskX!RoL+X>;!`)Nc%lGz&{1LN8mX& zp!@Uz*r@}){iXOjbL{j$+v)*&&&1gH6XSjs+Rni<_}%$|aiyaj)@o_&%+Fw@*us1I z&xuQ5Bc5APrr@``uHZ}=P0#Isi5Zt(UB0TUcEGsj_>DGi4sH&GP07K#j+^rokd`L3 zH%mQ&TLPN`o3%|PyxT;;?+@s03Ll~>8}}EgF|c0Pd8d_FZXm#pg#rBt^e|AQ3(qTK z!}a(tG=XOOu(o*i`g+v3;qgUxW1}0ofq(r{(B9a*wtT(W4Wpp#_%PP+deZZmy`IlB zQLF2wg8fe}bT*ZB4t)(YI_FF9mCy!~)hI0!`fp0IkGGy1GSP+Qm8;ikPVJ4Wm)950 zc|o(`h3A3k_ad+xF#<$cjb@X3v15Yk`Z#ozV{RBT*<_e1yIxycURkPb1*cE98)3vn z$K_2vU%A1a8@U@@uUdsQ&bOgk*K>Uw7vE?=P3g^zmdadt zb@gJc1(SgC_X{b5o1BzaT0i+QmS0||8e$GGrMKaj(%X%D%78g^z;0aM@wnfL-ctJV zO6|b}XbH5S2Mh^vpwzgRDm{1+(UV9%KvA9rPS|L7V983nA-qiocDU1lS*X0QQ@vh)111+3e4}OIG#e7GV;n?Jni)-d|U`xDcQ%(4RgK#5q z{FZ36ozUIlYEtlaaECjte+RaJIY7HStm^e5zt?C+&Mm%qHxR6DC)IQ}4L|TX9NVRk zdmGSevnxXP4p$?4EX?-~_teP#0q&qnfc2wGCDuZ>;7tAhL-WT!=9@be>#Y@iO4zt0>G04qNXx@o4H7 z4Conn<2dp?U?kbmq%P?$?d2i@Ty9#E5QclEG{mzJWkhDS>7w>#v6qi^%{|>*qIY{D zZfEYbGuO3&c4JSrlcvnQto2N2W2Wt&m(Ze0S7MJqqv|h(6Y+bI$igU9Ml}2uuUgqpAs}|XL=0L2d$9>{ zxzeXLF@ZPe)`{Vr{cJa)h+|qr@)aP;hOxj_MG<&ysBAB2Ie4IBELmw{WbYR|bkU~c z9XbkrfC&jCR$31dC8qDh@qU;76q2R)Y0r|D)Ds?76>2QEUyG=;wjmvhbZKH<@crNQ z3bL?c^mTSY7T5DKzmwV3;+FazYDqGL6lC(#9 zjRKY*!o~oz{s*iKdW9jqalq^$Yyz<25H<-|X$YGFY-9+V25fW)n*nTW2%810JcJ#R z#tIk0lyhBVrMXIsaFd0l%!dxwPPUDz{x-hda;>XFoMh%y;uy^-7ITXU7l z)wR{vuGT6`msc*YU97FHRaURnAXZl@q6)_KkepP++WO$+|4C7#u{E;@)32PW)L;cF z`{it9Yvwh~+xDT9LrHqp4#`lmlwKoPWl#qtkQ5B*V-+R)aeh!Cmh1)hVaZ->9~8vB zJsIlzV(hIS>BBx4{|-vh93YjWWFQ$BXcbE^GZ?@UwlC=%DEVyafNJpw4oDyp$$?2k z+~&y;*8d>(Hk2x)B9si}411F5xL5`YSJ@`_A%Rqxj&S`*83Cj?Dsxx35m&k&+=b}S zby>r2awmkmm5bxhN}NLS43cM&%p*C2WC6)@NM1mK8Jc(r$(MmtZSf7{<1;^wNP7qW zepnJqsPauDIKa9c4-8T{7+*__A&$BIV48zbBIrh)ZWMA5g&6VdLK%WmiI~);qhyMd z<7a|=-eX|MdR*4apz+k7Kbw>ce4{xD{rAfWU@&u!&-?PIkQce*@Xp-vT}yyLI03Ew3FL~Qw*fj zsq|}+DW~B~wBCoS(szk0v%)PZE->4N*1b8Y$MvPOemKg@X?Ym>IucnjBeOCGaSmI& zo@mH?TyxTzB@5D)MOl&~a#W7RvUVb4#kY&Sqf*-`$RqNoG~}EdmlM0$-Yj6Yq=3xI zGGZW|*)<_<%OM1spTZy#`zm=c5b} z1d+-M-te4GBYLYoo-DmHn3N@{TkMsXUDk_`Y42ZQlLQnUT4j@2z+VM{&r~*9#vxA@ z=Ros^@P;>l*wYrZz;nSf;TKX$@aIwRWee^pa9M!|{X60Mv_!K1Fh=_|{52jL`N4XO zePR{LWR&#dS{lOwvCucO>ZmfE_#X_%QI><(t**y`w;c!X*O-q&tCYpZyR;E7R3J{4lG<^(y?*lSPz literal 0 HcmV?d00001 diff --git a/__pycache__/prompt_generator.cpython-311.pyc b/__pycache__/prompt_generator.cpython-311.pyc deleted file mode 100644 index f5e66de1bd629765e7c9e74df94a576be7e295cb..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 9458 zcmdryTWlNGa%cI{l3YGTiKL#ZwAww_I8T+!GA|`U3aMq6rXS-`6q&8;Q6H1Lkaw@lK*+L%Gc8*= z^RkVzEH`jgp5|=4oonD79L+m9JI`{XG_^Z%8Amc?o&bQ*_ z{J`ht8b2n>9?l1?4XB=3A(zQZPt~QW8W4gVVurtOwjmeHPNl%uocS*)&VnYCYIr5& zD?gl#Yp6HwUEsV|YdrV&6F?ILr^|WvU zHL`=p@I~}If1Tb#Alm|OZM~;z^eSt6lxvl6!+%GuP~7Ry%%4*!3OX+uQ)z+!U@ax^ zoN7s?Q+XJama9_}v#KML1u7KlgPD}DifZn5zViE=eUG39ysz>Hb#!jis(dac4;xb`1b7h zLiAEPmrSI^D}Yo4+@P%Z27r%>V$Jn%WUcR@LKqtHK5?4S7J#X{F? zv1_)_I#+C+(;7gc6l&AVb&}seaYXPVz%EmriDZ&b;}Vm@RVa(?R~xvLs3SI`t*UKd zaeg9peQ_f{?n95-i6{39vO6i43!;bQ(+t zrGqtxSuVAjk`ie)AxOjQ=xRzx53^z}of~Ec2SsTun@wd`)9X-ZVoge|tfeCn=sbG{ zicO_cS=5VPNx+wCPOhw~)@xHUuPv@(g9y=HEh7ZRpYi-+^>6+fpFh4Du?U@zr4<>W-2_d~6hhdV>Ln~B!b}bW!hGaz5Q$gdaLT)WDs!oZ|0P%%OfwsfGh-M)zwNpqP25Kn~F0tKvXZMo_qe@^L z5{QFhr;*`B;%nPk-y2jqUQpW4Y3DHSCF1Mao!Gmh^juK7FKXv8?*j7i`@Z|52lL9@ zywbOTseMs<8L1bEuUoICYp-kn^#e)?ybcMf>Gf-`Aj9hv7~t4;uzu|-g)Ec6658q8 zo4-#T^nCGw(lv)EJg>c>lc$h;YzZxXZCYnp1eU&ISqA*tlFsrDunZilX~eI+tFv&x zGPw8h{w-zbRVDJ8#_PE&6zPjrOmc`8J*KHN>5EZH?=+^K8SMjI5=W6flr-}zUWZ0r zu{(qi3^mxkn5#3fv5CcrnHVf$DqUW=`34zpi(9sb;hx=VdpG}fWAcN~NY;mKlna@+Q>>(t)0{h317Sg~tt+r906*gmjl+wUk0 zzIxDEXrC*#&uzuFVvp*JHM^#vScK0hw&d;D?Jsx-p**M{b{{I3Dt1j_1&5w?g>Kj9 zq=0P|E&)&(uqjL|NrGKdRGD;cHIAMdI+?K~ZWWcs28ceNvM}Z`Y=8)jWMDi%G-ige zWS339Bb?>;CViB^H)pN=t{XM>dasfqwWC#TI2y&BBbyG{v5U1o^Hgd%THdy&f8IY~6tS6LpfoydY zolw>dvi6^-^Usm>K)w2@1=f8+oxLFI{E0gIK$iK5vVM?-)!BGXsBR>Zd7Du1A|=vN-b*Q!=HM_&N@e(s(LWhr+1dG->Dfj0`gm-7{uWGt-`sfv|M z{kN7oWBYd!<4w?MSFw zCL!fW%v+^I{K7m;c-5L0Qdvp0~P4aOOPI7pm54eRP z1W^Pd2+klFMSw>M!V3t_A$SSFc?1^#M4ZAT;<__h^_8A_q{0kRXA$T+F}hB0`T(k} z3QdqN;wDgOF#MKdbabzBYf^qq5}%FHYlu7+u>lSb_@V(~?^o!)+H@FvV~Z)#ft@Q# z=(Ldv^l2mO{Ei#n0^PsGJfdA&=>pwaq+1ob^^wcBJ-ll!xO$7OUX29p*Qq^AiSho~ zZ zn3*CoqcAg7h8xHLT>uTiRA45G%!I;B{CI{XECKAiWTXOfv&h_3n48rS-b984CHSV1 z3d~zY<}HPJtISaH&h1?)o|;y?b4Ge>Ho4qSAl!l29|0UF`EWepxcR<5AKxEQg5yRi zFxQIAHHEoW)#6yHHi9G87`Fy|tdHR*I;_qAe*D{Szx{p&gDU!A4&I;Wn;bHKIb@lP zHheis0ZiiTt~}_y=rPg%!f{$PCZaZ*>LVJ?%vtJp{LSXNh^P6PlYn3?owG_+{~Jys zntn|DQXje0*Us7S)L=8jHEf2Zs<#Ryhw7y`dW{IZQq`|S4$fX*)@F;`vRk8F#WY9b zDX>UwO~GSo@=H~Z4r&U5$L09Fsh;mQl3?wD1#8`_T5PjJCc8E5 zu4n}9YnMA@i`*(V%7MEOncb^KXPaS=sm1|MB|yd^hh;xXfNbb4Sp8nr@)VwMRqL&j z3fJ3kvlBS0{(c>2C&&E3U2my(z}B1WW|z!z9=R(;t(sQh!U}v+Z`CKYpif$Jr{Q~g zOq17dhQtj^RJtRn{e(?*!I^6H{)smZ;n1^DZAL5L{xO&3 zRGVQ&u^UJ=F3odL-O~|X2i}SoUiP?iLbNh4wt*@>fpb}KZ=%HBt5->#`-ajn zTJW4Ldd_ZH9tJ`?ft`&)V5k@v+Oj|F=vI1O`cL!yvHzO<_sKs_{l`?{!bI`HM4@A{ z*fIIV%|gclcsr$V>z4bG)2I0R3eNtbvwyEoagIQG*cenor}h>KjibfJ(XFxauJhZe zPo}n}KwvvkaJChlZM#;*2_k(D{cXE;rTt|i75o>A{)-B8@v!&o{;dbSSCrl>J7YUz z2jdD2r{$;i?mS@6EA06l%Z}y#(w6H{pnWH`*Io#mF9yzU*&hXaO06BGU>By2fxQb4 z;6kF~g>7z|yWdpm8Qp*DLCZkP+D?^1Jtcn-Q|D-@>s)E*)q@KU zhTc$y-q`Kk?N#P(YOU^&<1qp71i==gv81KLICCs9O(iB&V*Dkhy~MOadhBXw0Y(5@ zcwu9AeMN^J&<0COiGhS%x=`n14{Zlt0GbX0#NYgB`(L#0{JP+a7JbnzTPfK3RWSM> z7~T8uU{MK13&F`^aPqgzmSt%#SZF&Bu&eTa~`H6uXKS3o3o} zwZ-x4GxKlPOx7d{Sl)xhKs7_m@O1L(!}A}AaTbcVX9BL#I=}^0Y|)42M=(6`Em^&q z>5>&p0N}Bi*nA)V(#Vi066pJAs^kfObmP#|_0bK@60~+4HnwS2_yuREu~Tb6j3&N- zW=G6H+E~qr7(-gxH5X!T#5{<3iNmG&5F_>AY{Ytak#Zc9VO4eo74Lf diff --git a/generator/__init__.py b/generator/__init__.py new file mode 100644 index 0000000..7544ad1 --- /dev/null +++ b/generator/__init__.py @@ -0,0 +1,38 @@ +from sys import path +from os.path import dirname +from subprocess import run +from importlib.util import find_spec +from platform import system + +os_name = system() + +path.append(dirname(__file__)) + +def check_package(package_name: str, install_name: str) -> None: + if find_spec(package_name) is None: + print(f"/_\ Installing {package_name}") + process = run( + f"pip install {install_name}", shell=True, check=True, capture_output=True + ) + +# Check required packages +print("/_\ Checking packages") + +# Installing from git because there is an error with normal transformers package +check_package("transformers", "git+https://github.com/huggingface/transformers") +check_package("accelerate", "accelerate") + +if os_name == "Linux": + check_package("triton", "triton") + +check_package("optimum", "optimum") +check_package("onnxruntime-gpu", "optimum[onnxruntime-gpu]") + +""" +# This package is for onnx models that run on CPU +if find_spec("onnxruntime") is None: + print(f"/_\ Installing onnxruntime") + process = run( + f"pip install optimum[onnxruntime]", shell=True, check=True, capture_output=True + ) +""" \ No newline at end of file diff --git a/generator/__pycache__/__init__.cpython-310.pyc b/generator/__pycache__/__init__.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..fb78b36255a2a6f437f83ac7a0b4677fcddb6d56 GIT binary patch literal 997 zcmZ8f&2AGh5VrR}+fC9IiV&w^-uD_L+HB|Zms}?$3W^EFoFm! zQH4{CEp|$$a#Od(Zs}Ei>bKY{gDOnJ7W-vX_0k?jw6{TNO!4O(>e6I`zt}ed+9x4U zM0$&W#AN54mk#KVfE_#rq)!GLr$s|DY*9>hFC6&hG(PwN z5u-6Sp2+nI&X6$4^NIp=Wy9g|K*Eu5@0OTPcuK zHJ$+0PEhd9^RO2loq_uhNs5OfaZ zgpmES^>CWaMyK098DEapiC|*X1u=Ryc8#xAv@DHZtY~pz21PC)B%0NY7L6Xe(uW9) zFC^nyLPrs(LO5h(^`Ve;L6!Q7iOxcXq2w2 zjMs#!8TeH~A1*1UGS{^{EbOg@&Hia-S;jciSvC<@rr$=Fb*0>auT%mo!2|37#eclt zat{>i1Do%$O~q#0-B%{jGUsYh%Zf_1o-UcbztUQ$qp8KKWkbU1h++A(CJMF z6X!)iOB*RQ{!_-AOA~0xbj{PS7MfK}wI0@-U&0)KL=TsuS>Np}FaI;V8t)nxhC|w> zo3K_BJ5bWL-NdSycO`9$Z3e6owbUh>Pa4h2Z8suO=GrDB?VQ*rx^c>dpq!W}%NDGp zSqA@T*%n6j;bz;T8*JBlM-Qu-G$nn6K~f`N3DkAF183meaiS=8V;ng#j)6SqFRzjn Aj{pDw literal 0 HcmV?d00001 diff --git a/generator/__pycache__/generate.cpython-310.pyc b/generator/__pycache__/generate.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..4ee2b37edbbf88fa07c1234c396de0ceb4b6e63d GIT binary patch literal 2279 zcmZ`)TaOzx6dundnM{)1Y&N?~DU=bnDpDIH`hrkJi`r5lp-@q^52KYVZ|r2Jom*{B z)hy{tx3BOkc;RpGGyKX^-;j7fLKHa1$);PDI+`z^kI#6H&&_4G8#yp;KlpPDH~@Dz$()=dgdle_H}?`R_Y*%4lEBKmtdTd9W*#P?o%>lUkCKQug4}j^ zz#GpU-Z*xTNz%5GCJ#XpT1khu?mENh1$@aF64T;Lvq{F38hU0h6Pj`{Wp$=$l~y84 z3t_fUS{B8VSE36jIuKe5sU<7aw3NA!>-C|#+=@*mA`_D7Qc^vygxSOQ9y6J;gG^AK zPIOupOwPfO@XqRc6SVB~@L+Ns9`y@=a1z3u#JvYzN<866?j;S;1nvV5lF-T< zz?(_S^3du;R;Oio+wurFNjg@iZF$%74)AWWAvR&p4dBhBXXTq#zGdY-;9E)G@;>k& z*|s{{z`bN(_a9jO9jmhgd>4GbXF9h6B%9(mW;8hJS}J@ikUuq@hQF@=nAIs~a7d~qEED75t_dkESS~0v z5v6&_Ye*4lI+Q-DS+-n}Z7|VzX`#&~rK~7Q&Cmu#sqEmAe^8d0>;jdW2t9-?gg(MH z!T`Vo)2w7#?x0{7VGrRggmVaQ1N`jB_mLy~dH?3;`*$_w?ag#LtrdKA|3;Zl=l8ze zpVZ*&oE9Y)YQK_YUg^uTWr$08d4gJXy1$m3Ffpp;vIU#tsZ8&fM4UjXc1&xP zs-+woA6o*X8XPmD71M{Nn<~04a$y?FY@J}mo($}5IKKgpx(~2$IjJ4l*3N<)lZ8W} zp6@!^#o1GD;c@q&E6;0hLAdwBdL86G$S-RDb-50Ba7><&VPpK2@v%!mFIkz0Ig@Oi zX;Vkv-(ZD~q0d2gi`V9tGR7ViAI#%vnQjTE^CI!-hm}$d3>j2jpU97^Xr6;^cn%^w z^v6e|lO`XhDjt73zV${w#h72qX1B2_9Kme|9D3Gyrt@80=rmvYOJWTSeR&D~vT%s-nO?4tk*itQaxWL9J^k|nbjYVe*x$hQ^EiM literal 0 HcmV?d00001 diff --git a/generator/__pycache__/generate.cpython-311.pyc b/generator/__pycache__/generate.cpython-311.pyc deleted file mode 100644 index 73c90582083a325b2913b215e1810678b7fd041a..0000000000000000000000000000000000000000 GIT binary patch literal 0 HcmV?d00001 literal 3370 zcmb7G&2JmW6`x%WzeQ4_B#H`Uc_qhoC_55R%RveUje|%I+|-g3mRo3HmZ`^QWsg`oVk`ZwcJ9HD=3qF>oyHR~jp;`8q6)YP{uq8Vm|nXaSLCLAQO(c7;w zxD|9FJ>ux^(0PCeQY8ba@@Jq^H9|0n8uAA)Vnoy!i35cN6;%^rEGDS9niN!`Cz%qG z$sUy!R7y~iIwB;~Jt`xp5kY0tEXlzRSwY3sQ8AY5jg5)1Q9+HVc|qj`6;;QDV(*DqBn0+NKOi>0Ct%QMm`}9R~h+;*1Q)ITrQY!8n#9WYfwk?$d?V`RET#Y811pD z4a?FPxyLB?jv!^5j%E?3#%{m(UdYhz^%y~x=+tUz9&_t;)2V^^UCq<&xXXhXOzlty-?m=oDuZIZShSi$jUS48Wh!5g_{Ri>2GHO_n^95^syprCZI- zx?Ukmn~b~oidn5TJUGPC$F5y%es*)I(g5Atn&TS8TdGsnuCsS*p~YSLZiP!V%%uUR z-y@6lCQb9rIq<)S!8cm3{hldvJ}w@d^D(q@Z9LaGGw0JV(^ozhHe zeSg!(Gf>-jrZYX$`gDKtm>wH%t?pfEt+&>HvECUk0CDwrvH~Y997KH#?YTBS*Ex0m z`HMcDK3MiKv=`g>VrS+ooNv{~XQ8$QLWiF3t7Ry2KvovzNQLH2nS;a(OyWk=kg72w zwu{w-j7V}gzg8t94sr>SG@>L1a>?DOnldDwNrPlBVqLlxV4gK01i`$PR@d3>AY*#k zAPYZvi3{>e-g+D3ER#U0EmO<#E0&*9b0%C@;$d`t$MPQUF zoVu(mZkdiCPwla&Y*FxRkOGBX;Ny_dOvhxJww)jN&tji94-(!l0a~b&E*`A< z={cxRiC_u6z&bomLq5#aB7ArkI^e_T3uwdA%CXnH-8kXNUrR&2D-V0h zSF${`){az_Oo)&_NWYRozvj6)EJozG4}po-wio-~6pHC7YwK&P|JBY4kJa$~%=VHm zF~p-n)!n5*Yu9Y)EdhmfuJj(qr z$o16j(nS&DVT2#YHGJ=iFEAKyFw-(wGmLz0^I+t)?_G}y?Fmf@tqg1GyTk|Km&y<5 zM=+J=YwuqG-H0Sf9W>_;*RG5uxr_cENFnJ*l8-(;UH`-KF(jwEa!yjZ=rp|KyjL7v eMnGW8pPt4`?gxPpm>3lbrDtVQ|c372R@^;sB2;I+*R2F|$VEp%Sp^`XL#xG{8evg^; zcOUkzeA-OuuwfRa%6Oo31=)yEve_WDHJsZx%SRZrGiDQ1E%_NgWtxA*-8Xj1wJY5T zn{X-4+*5brBI{b;h!B~bGt-pnQ#?|`G%PMwwK~?DhbmJL>NwAOt`Ryck4(*SF%3Oe z(i6ZjrCE(2PlpB4gAip&a1jNfbuOSz>lF&C7>ppv=MJpn=3;N>>0nQX2Z`E=qo^t| z+h99Sqmz$s2BWIfcxs0F%Rzw&iUp}0Hb?YU6%S@Ka~N21Tg8dlI8=HlRTNf<9?lIW z7Nw72g*Dj<7aVTl=8Babg4W8%xX{kbynk#Qi*ZddYbBTNnV6X=`q0L2jGKc-)g4?f zscwa%kxIzW9C{v58n{7$oHF&i&~aL&rZJ~mTKAw!uaHyIut6_(NnW01#|v$AP)W&~ zsLkU(^Zp0{_Yt^W-xGX#eXCfFvv!fSg+{Lr4g3!10#4O-50uslu+RFa-@4Fgrs7m)O~#$ z&fLRuW!cm=G{{0?QRogtx~qt^OTVvGXxVq_M-!Z`@wpWmQ0DMj800mfb0>Nx5;8WC z;{D9zb%uM2i7E^Iavj2D4y4It$R{%Zc6FA-b{1h0Kw~L?Y_5V{an0^s zK4iAM@}2s+m|mg0?9!?ed%WDIkzIg$V7j7v01fiWy4JQnZC-p~w5;NRRcu?u#+R?M zSK8URoz;Wvd^o)7igSM z_bCeBpK9E{S62Ivh-{4rr9TT&P*4i4y02}6=ib1(CkB0yjf1n$LdJo$322Y7W;&G$ z^-vHY%Nq<+#-2W+K_}97)OCbMF1q> zYTL4#=>zL-+q!#J{-@gJ<(KB{ALeW)GqXjT>n~_mBi4<7h-w*c5E={JB(buMhwMYx z^K44CVCH;E{?`lqFSN)P>gxN6GifmvnF;q~7J?*ba+z{Y89cQhs)o#`j9r9DUQ&RJZ{AQ$g!Hzejdth)qX^c}!^zm5~^bjEX?sY{*do86QoD1ih4072tJ zg7O(*jRX`Ohl1K-sAU;|nCz+S&QN3+qR92fQB^DsK#Ei=r@`*E)7{hROiByEt z;!*N*_&vlP{es-zX5bLMNvBBQS0C8Yw}U6UpwTp*2&s63?;j@)Fohco=9^V erjp?CNtb}xxh2hMxM0Fm)Foh=YtozxGO)Lm*|?AZ diff --git a/generator/__pycache__/utility.cpython-310.pyc b/generator/__pycache__/utility.cpython-310.pyc new file mode 100644 index 0000000000000000000000000000000000000000..58b58320e6861622bd7084d0309ccb7845d00662 GIT binary patch literal 920 zcmZ`&%}(4f5cW7uNEawnLbU3k=SCGP&J{vcS*hxU3qqApP!wVZOp|!A9oj|N6TCrR z0FHenUwi5kRH~ruINKjUs-+n_^Y8C-)Z*fR;CgfXjlZLWeDNQ*7sHQNC}tfMCmbqL z!yW+PRM3+0fU^s>7w}Lpw6K!#h|i%VQ&0@(Cfzj!$}yN@BwPa!RK#r5#-2uvcKHg$~Yf z@1Ucf=n2I_;Aoa5)D4WywKX54J#ul8S5akrr6phfiReG)PQLRk>TGH0aO4`YA zJaK~!E(?nx<4u1Zvy3Vt5siV~L;<~F5dA!RaVTtdnrrFXXFdVD=Xx^Ss1#0aHi!SJ z;LYN!kI6C@N-JuUz~20!JQYn=RJk$zeP(JNySc~zsqVQqk)H=*8vW2qKA4Sjdvu2h zALgZ}?=>DQGI0)5;5`!~vcEbFy5JH?y#l?$gDC~lArrlgq3szD|JY7q?@6L<#~Rg$ zvv8=I=A&a?pRwVAY}}&*VXe?s=Zz_qu7ysh#=7#PAZKuv*#gX?~$^qMjK1l`+7z$P`R2q3N>P8QI&ecE!xd zjf@n0$iWGQ1PZl}KBW!mAwR_~f{3Msl%9IqqW0ob->fC4a+1gHH}B1RZ{C}E-%nFh zlL+KD_L|)y2>q#zkiP|eXBjG9Q;3aw27XB_VU!KoUx%c zSsg24LhJ?j2Q`2_h}x2>RDTi`QSRy{P_DP3?Xr zyK^@&#?lA+Vd2d(PTGOv#HO!!L&9Tb*ZoZ%I$q#PDfK*xVhS*x|3KeD&stMgUITze z$*9ts!qr}({JOC8YhmfAaHChak(l87Ti@iL;dM+)8gz}8tEhTs0Kd*>>+hZ z4FrT5r$ZOBlv)*Zf2nJ9jQ8`Hk?|}+nb9Q}g@q4h(04M-Gz0Hm?+~D#wHnhf3i*~< zj>7OkOf_!Na?KCp^EIv%7fQIHtVg23MTrP~p*X`smeRCD#jM^i>wdsxJnecO4^H^x zsCJXbGm4F0kDmJ&s}&{d0gGCZr|z8Upl0-Z>BFM$TDYA z*rEMw{^`vpH}~g{vX^_=%LyXqX8IprJ z4(Uhf{8md=Od@%5KdMQc(jqpb5@~NQP#Tv~={U38h*&G&E8-LARu5Kw1CSUP<35@@ h9RCu_#MaN11i^9^rgL~^{{S4~QIP-u