From 12dc3791f1af7d5e1fd8e8e71744abc59816aedf Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Wed, 8 Feb 2023 16:40:47 +0800 Subject: [PATCH] Update Signed-off-by: Jael Gu --- README.md | 13 ++++++++++--- dpr.py | 2 +- result.png | Bin 0 -> 5740 bytes 3 files changed, 11 insertions(+), 4 deletions(-) create mode 100644 result.png diff --git a/README.md b/README.md index b87ca52..7250308 100644 --- a/README.md +++ b/README.md @@ -33,12 +33,19 @@ to generate a text embedding for the sentence "Hello, world.". *Write the pipeline*: ```python -import towhee +from towhee.dc2 import pipe, ops, DataCollection -towhee.dc(["Hello, world."]) \ - .text_embedding.dpr(model_name="facebook/dpr-ctx_encoder-single-nq-base") +p = ( + pipe.input('text') + .map('text', 'vec', ops.text_embedding.dpr(model_name='facebook/dpr-ctx_encoder-single-nq-base')) + .output('text', 'vec') +) + +DataCollection(p('Hello, world.')).show() ``` + +
## Factory Constructor diff --git a/dpr.py b/dpr.py index 004626a..cf84ffd 100644 --- a/dpr.py +++ b/dpr.py @@ -50,7 +50,7 @@ class Dpr(NNOperator): except Exception as e: log.error(f'Invalid input for the model: {self.model_name}') raise e - vec = embeddings.detach().numpy() + vec = embeddings.squeeze(0).detach().numpy() return vec diff --git a/result.png b/result.png new file mode 100644 index 0000000000000000000000000000000000000000..28e39185ccd043854c79aa222d23d336c1ce3846 GIT binary patch literal 5740 zcmZ8l2Q(baw_k)vq9xHS(FGgPSWL&-hopB(sN#$2p^wI*p#!Hg3g9F?Rk>-i5n|~ z1C=2(No05cSowcNXoh1T02U79#0YR+?<4s~`#<8pk^kv;VyhIsX(zt4rw?0h*O4gy zfiazG^qg+=M0=iox{~ia9EXB)%Y!SXbUz)0WkYI^K-omc*^UUZ>)R4iRCR^vyNUDi zh;!$o%;^5MDBsCl+3f$6Q_F8+d`}tBZWa-4e|J|P+Y7Wb6LIDrC>SyVj3icj_CT3N zqkeRo0sA)YS(NHUE<7&Ne*!pIa}UlY)aAwx;6yYW=gVRJbXgu@tR@lu=CqVEhhja! zxeD+#DehPKisa0edst@jwbQ%EI2#j~yDv^b3eQB5S7wd8$kvT{&&t*(kl$R4OWzJBYV~cPA^G7V)XBYGhg8A&q?18>V0Sp# zT!Sey)$SLMi(u6Y=uDA!5c^nNdnEd5(+Q|}MYC}+&7D_cFE~;%<4<`OWg&BWqMd6u zH@{uM0^;7%YDkecrmqHkhSMh#uJC64IX`w6M|4{mheoe|KTyHKg^q6ZeXCFYpvBC+ z+n6zN?di8u5eH+&#A=H~9>)@_6|bN>0Fg%DglgFRIQTr0Ix%yyx$fJKYkSGZ$h;!& zt=TU)l6w_n4~k@S_{pl`<*eScJaM7!Oq(r6M2O4K%?L}x))8fk0c=h%hKy+^(=K=N z03l|Jge6|_`SPFkl*QG9;0MaTFOV1GUFzhH5g%^4W3w>OhR8y_$&NAcAG=?2r)-yD z&lL=pK^)0(0e~fjv$ckal@Yt7J@74#ZF~q5+!{9NfQq})%{;t2 zuDtXboX8`b-lsa`SK7j3NY`$x_V^TYX*gSma{9hRnblZjW%>B=P=mHVB;r z-%V<+I-pfrU$g&HM?DQ7qokx%Q6{}Zcl-7YKCY;TAw{D|3Zty*kfG7>O%Ds!_Dz$~ zD6{piN!*i6*iqH2Ax$gy_oe28PGw8da_wQugDr)=a_8^Mey@)Go}D=im_1d@4(5*& zcrRU&YL9MNeMEp2tQxJtjK(rTsT-yDztfC1oJ-^34S2@xtf7^-1K|_F^nh%mzQadd z_mHsWOCLDSd~riS-3c<3mwQLaFGQ7t>EOc!lz(o!TO#0T&NNP80Y`c4?`p_JcKTiS zWc+;4*BQF5Sdg@RcO%`kZf_kdl~esNRCsvnkxW@q=eesZqv+I+PGnmTp#o zD<1AY&J2toYkK#L3(k$ZY^w?j@trCJ_)Ychd(^o?>FI)Lh%S?J)()mDd$M@1Cx|dI zt6x-qJOBlrI~g|L?!5WA(+J9)ulZgWOJ#5==N3$F=az_;!5VVsJ~`kZSZDPxlKHsy zp84J)(K-+sny>(ji7w)564>ONAyRDN!Br3ot+Y=z%iua$Inn(30D$VD@DSc3Gb5dM z^$r3RDlEBQKmtE%rEcT{l`tED0BVD5Qn_)7Aeox_DJcZ?7pVVa%<~wcq-e4i#P{T z#R7gbqZO3t0J+*(HY;5skU%L=;+_8O*w1P&bS%`<#zU`;XLbk0jMM74rkuh%PA0~= zQyA}NzR={*I%WM#j;o9&mb5l%?TeUl;j)Wg^QjD~FfF~!wCu;r)45p~`)6~}^MI5k z=#UGz?ZeMFN`9F!NH%ckkN>X}_LCm@;309YRm<4v%jQj69Bt7bseQ&(oj4;F|4!k4 zXHX=#bd@&WitD#N8?IxFSd$9sYS0cQ8+_?@OFr`_mN2qBKC&-y&-89TwEv=d`z0~T zZxR*8P7|2UNgM`U38AQQ8SiQxR-2Jrw<6SBsZ!}Y`YW`{@UD}gAcq+9X)7DTax>!JsRr0 z#w=Vc$)mx>D4-E#-Cu_i2e+;LiufIh8}@sWLim`Z!4|U8`bz*)?|*<65<+hfQBIdF z&&uL79Js3&@r4Ydm_9hSr6fyGOx`)r-)d1J7J7 z474CxqsD>+L-J%a@44`oUP$o0RXX5VH^B_@q0Q8!PE>G}^aA#TwJfU%xQ!9gSv~PT zhK0P;s}-~yucqJ?(6OOu~eT`I}M9TZ7YD zFb!`@_vAWLck>Ozhv)BCoUvBVmD>RO4F(;G+}?;dGe2-fKsbVy~IlRRlbK@P6J3MFgA zw4c-&f^O6%TD_Ffe0=F$V65^!ia1z1^F>bnG#+1CqMG(h+5Do1m4I4O%=0s+KeB11 z%;00_V7-*X@WTyRSo3Xn;%ZIPSU3CLk%tt5V(gM7I4M+PzOc5ys9a>Io~IIM93a>M zc@0Ke#4BP!Fpu3xPX1^HiSx2VV;lVuC`_0?ty{4yd)|VdW?##f!_KYF4*XVvWiGnb zZ#&BkqNi;2c9auh+f6Qvxs5xgMS&ckTBG3$;Qi`MH7i1@R!e1h$$bEsaY^dunIscY zC2)erD9Mpm8n9YLuUQXhC?W+)lQv>nh@xL&XAi(QH&w7Or={?)GzQ9FUGK+Y+0)RX zBu0n0GZ={X*Mp}rD(@QDa&#eIihLyJRCzaFlNaj3yTelT5Z9f`icIU9?{ZR}B74t0i*v1R>%oc} zw_F$KguyX0b-J9d!$@Y_mi$4#wO3!fPrWjDc+cgC`vHa%WR~nA7NVeyF!|=xa?>(( zz$IhE`bDK<%89*x`|?x_*n9^!g5*yoCJT=>0@?GV3vFD*83%C zsd5zRXIH9CRMuT=K1D19*LoebOg6DSW{RGbB0lf;WMRIKAbu3*)f-oF z%!f6QxEx>Vmdr%Vu&-~}JFNSRg)uGSpQD_q#a>YkJVOh0!qfKd_1*SK-mXV`l)FyzJt z($Je^Un%GwNNjnWI1HQ?7iyNkP|sA&yXNr33&aFfL?DNTt{Jj8`^VZua=g*$;yH+$seb(|k%#Dt5XY)M#P)2z)IJH3J_E>;PB5E-OYff^hSG z&v7kTCWN2RC5%=X6M|of#=K{LRgX7&Tn2I^w#qYZn<*KbheJdNQmYEhEHsmMup`;1 z!dq5zTMv~Hg`YvtFQhkIt>b`$i>~+;-SlCYK+na?ZSopFWiACGa!3D3CMfz-zt5-q zg@REu74_)PFZ%0yO0&d|AOX9kXWgd_f5nQxtBvD_oxi*O+dpaC%1;aSnE6K8h=h$ORn-^2wt;GKr3Z^MRORrg!K_>k6_<{zeaj~Of3M4 z?`fUpDL5h*3=W~~DULv!q@um_Z17i~KeNWE1XxBcY6%WXW9eo0Q3>ebNOmUCZi~yX z090lzkgq)B;0Q;vtmyhgVzQt|ag|E+ zEn;tbyL~B^6x$r*n_%74_2!Ltu#H~0{PW47gKi_q(N;V7NKw30Lq%y0WQ7QzA$|$wW}QB z4zY&!orXF7fsB(BdYKQyWfo-M_=w6Ri!VlKq3A}>R-^zQ1B*mC3nPFDZJMp}z58u5 zzwK*SAZ-2hCBuM{^aXR;Ufa?U+_r`P`fE1J^GQPt zy4A{^_Km$USghQ6+dV-|G&jK}_)&dTkmG?Sb&bDH*>@eeQPxP|ZK@P4RQIn`b=#=7 zGpsqll05vp-=#ZT1sSbUaz04bYrH(L=nw=erBY;pu#0~)96hf&C^o#uR^S-KYHEwy zb1B`k@V{Z>E9h;uknaC?S?JMpI`xmj(UF_*-T?>8g|EwX_`ZB{)vw9Lmi8W1!09}5 zE%{r1&w=+hjsKbf(l_W8KIpgIwi#JzVQDR5;v#&~3V7mZ_-rd&+PlT8f2K2#1Ge&* z6QJXb0RA@Mya{5ht>n#TkybGt<+SrwvRD;CWbYG3$Tlx(`hJ(vfhnl%`qr~7=YL#2 zL+nz-lEnA&c>=geh(%}YF44_sF*D)#XHPt3TY1MjY24bFV_EE~jkVIM(Oy=^^A29r zdNc}u8BHYlA;Am6=52$o^wVVG<5#gz(hg-G3qx+U-Db&_5%Tywc;#~*?YPqfIl!i# zbE8M(eSI*&wYIXx*7xe&8~`52MHIlRfJ{`X1=0z}&$A_a?cNzUaM6+Us}rC=k>g5M z>y_wDo3NIJ)E>B34s0cbxnooW^8WrO+jf!FWi2w*oLnUOegc65zUEw^;hY)DN1D;< zQ3ngjliX}>=|K$i?wca+CoplxP;3HXHr_=hC2H}6bE?c{2 z=lvho10UddWvxi{?Y(P9Px8K{P}pHv+};Agk~NW%9@R%uATnLVy7&9$!`9KZ>mmi| zlhnDnJsxVp{2>J<2azI0Eej4)19!i(cKnH$aLOolFBx+AD zVBJvmNW}4_^kni&c9< zKeV)er87=0hz!mw3a?&F48T6;np!IamOiqb_<0y}TFm!hRr_7H2;Wx)#d6RynnV1? zy7o-IJk=|S#fJedR&0Xj)}ge=VjLr9f>wyR+-u>q#y4JAD7K7G zYHs$T1J_VWH#{zX`2Dc*Bql85{iR1P;syYq44~Ne-f)XNreyz@)a}Fmhrdnz55@c1 j2nNtb%>94H7Z0GoN(Z`Io75uyp^+hKx~f$wc9H)De?cD^ literal 0 HcmV?d00001