optimum-rbln 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (29) hide show
  1. optimum/rbln/__init__.py +6 -0
  2. optimum/rbln/__version__.py +1 -1
  3. optimum/rbln/diffusers/__init__.py +7 -0
  4. optimum/rbln/diffusers/models/autoencoder_kl.py +30 -9
  5. optimum/rbln/diffusers/models/controlnet.py +93 -23
  6. optimum/rbln/diffusers/models/unet_2d_condition.py +78 -61
  7. optimum/rbln/diffusers/pipelines/__init__.py +7 -2
  8. optimum/rbln/diffusers/pipelines/controlnet/__init__.py +4 -0
  9. optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet.py +768 -0
  10. optimum/rbln/diffusers/pipelines/{stable_diffusion → controlnet}/pipeline_controlnet_img2img.py +25 -16
  11. optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py +942 -0
  12. optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +955 -0
  13. optimum/rbln/diffusers/pipelines/stable_diffusion/__init__.py +0 -1
  14. optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +23 -4
  15. optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +22 -9
  16. optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py +19 -3
  17. optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py +19 -3
  18. optimum/rbln/modeling_base.py +36 -3
  19. optimum/rbln/modeling_seq2seq.py +19 -4
  20. optimum/rbln/transformers/generation/__init__.py +1 -0
  21. optimum/rbln/transformers/generation/streamers.py +17 -0
  22. optimum/rbln/transformers/generation/utils.py +399 -0
  23. optimum/rbln/transformers/models/gpt2/modeling_gpt2.py +24 -333
  24. optimum/rbln/transformers/models/llama/modeling_llama.py +63 -45
  25. optimum/rbln/transformers/models/whisper/modeling_whisper.py +13 -3
  26. {optimum_rbln-0.1.0.dist-info → optimum_rbln-0.1.1.dist-info}/METADATA +1 -1
  27. {optimum_rbln-0.1.0.dist-info → optimum_rbln-0.1.1.dist-info}/RECORD +29 -25
  28. {optimum_rbln-0.1.0.dist-info → optimum_rbln-0.1.1.dist-info}/WHEEL +0 -0
  29. {optimum_rbln-0.1.0.dist-info → optimum_rbln-0.1.1.dist-info}/licenses/LICENSE +0 -0
{optimum_rbln-0.1.0.dist-info → optimum_rbln-0.1.1.dist-info}/RECORD RENAMED
@@ -1,28 +1,32 @@
1
- optimum/rbln/__init__.py,sha256=MWckW5I-PQYH87qpw3-I3hMa5cc_RldVOf0vEmWVuJY,3741
2
- optimum/rbln/__version__.py,sha256=L6zbQIZKsAP-Knhm6fBcQFPoVdIDuejxze60qX23jiw,21
3
- optimum/rbln/diffusers/__init__.py,sha256=Ai3nNlgBqXnt90fv8p1AXwDKFGIeTpsDP67nMJm8Ydk,2295
1
+ optimum/rbln/__init__.py,sha256=1aJPifD6VJCf5FwnA1-dxsVPiEqnHw9ots5T_aUXm3I,4051
2
+ optimum/rbln/__version__.py,sha256=jZ5uWOiyV1Hb9rhNyo3bAgfWmHx8_xkueUWpcUrkz_w,21
3
+ optimum/rbln/diffusers/__init__.py,sha256=JWeu2ihHKiYD0Uzs9jXbaAq-bA1G86UCMPPx_oiJYFU,2606
4
4
  optimum/rbln/diffusers/models/__init__.py,sha256=aY6Llq_31dZjdB9HPBDvi7sXVtdQT9r11gokXG5ffxA,1139
5
- optimum/rbln/diffusers/models/autoencoder_kl.py,sha256=7Nn7JG3HFuFJSzTtNVoVeittUM8FybHaaJAY-6eOoz0,11775
6
- optimum/rbln/diffusers/models/controlnet.py,sha256=-tHH1NAcjCBpltzj_g-ycEbXbW2Ptxd3F8k0-LfZkPM,6262
7
- optimum/rbln/diffusers/models/unet_2d_condition.py,sha256=pAVNoW5mg4DcZDhaGBRta9kj0scjLizxpj3T8HFgjNc,13327
8
- optimum/rbln/diffusers/pipelines/__init__.py,sha256=qnw9AXFA72KFLFL60FYSQ1PfjYlA1SVWvs5WULETc9E,1300
9
- optimum/rbln/diffusers/pipelines/controlnet/__init__.py,sha256=Z_YuJRDhhOvIA1H4dk6LtOGCjGGurwH6z4mk7aS8-RE,1047
5
+ optimum/rbln/diffusers/models/autoencoder_kl.py,sha256=ifgsAoqZE1dOlJv6z7HJv7rp_IJ8KXMEmI_LOg98ITU,12566
6
+ optimum/rbln/diffusers/models/controlnet.py,sha256=7T5E-RvGawT2uEtuJYxGTrzIDbApcF13zuXbVCcoQVI,9224
7
+ optimum/rbln/diffusers/models/unet_2d_condition.py,sha256=dWDgeWhj2Q2LXHFy7iCnmPSJA4rSMCLeqwAIkYa7Hnk,14422
8
+ optimum/rbln/diffusers/pipelines/__init__.py,sha256=Xr_bQbpbC5HbJB2NuUcVQu2BGebDkc2bhsGJmL6jgps,1449
9
+ optimum/rbln/diffusers/pipelines/controlnet/__init__.py,sha256=k0govvSBxBUR5qpxUGxRMHuQCMX7hXHVZ4EqVRw1LWk,1377
10
10
  optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py,sha256=y-CZhtywVnvnWloCqKaYVS94_QprslZpA_EFEY6pHpU,9466
11
- optimum/rbln/diffusers/pipelines/stable_diffusion/__init__.py,sha256=A6IgKzHW-jVs-u8mTht5flE9UIEVZ-l9LR7jzQvUaDg,1228
12
- optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_controlnet_img2img.py,sha256=sYvEI_FUlAz0-o49My7a2VqfckSSF2mm_VgrePwDlsk,37367
13
- optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py,sha256=JSNSX6Av81vpB2FkNF79kc6MEajeGMz5Az9FbsJz4OU,4350
14
- optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py,sha256=H_45If9HnrPSvFSyqp9-szxZNosjCQZfFFQWKIYxG9o,4776
11
+ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet.py,sha256=qwpFqmhuNvxonMwtGyZU0d2bvknJloQRk_Sasi9b_gc,39622
12
+ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py,sha256=iGJQM6JltALuqSamj9WVojDjtPlJXbJeT7RQuu0Hu4w,38127
13
+ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=qe2ZW-ugpnfatx7bx1a21C_40kVMSp8DsQ5fl2DFoKM,49849
14
+ optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py,sha256=DKai4mdW_fkhdD4yjIGKyKv00COFH-Nz5ux9jmatjWE,51196
15
+ optimum/rbln/diffusers/pipelines/stable_diffusion/__init__.py,sha256=qf_uMWSwD-CyRMRC73y1QsTMyl_qCMreIdg0a8rhJuA,1142
16
+ optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py,sha256=i_i3729IE4X19gTu163hzJZqb93pu-VaaJj_gbzoLG8,5195
17
+ optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py,sha256=ZyaEsC7uGa9W8uV7U61wGd88JxkAy78FWwrx5SekHOM,5440
15
18
  optimum/rbln/diffusers/pipelines/stable_diffusion_xl/__init__.py,sha256=8MDMHIVsDrM6lZAyvpjFtWOFwiY_IoSxzCQe-gJYTPI,159
16
- optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py,sha256=vZDAg3g8IPR16tmhQVvH0anmWL7q2fnTarlTZC8HhPI,4633
17
- optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py,sha256=OzAHyxzpsAnmo5PL8At_yNALHH-GD9Q7KEl54dS-P0I,4783
19
+ optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py,sha256=RZ7xHD4Yo3Ls8hH9bGbjokZd7gTPV7Dz7Cj3z-_t17M,5366
20
+ optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py,sha256=rGJFKMo6aKbJwwLTS-N3fkFUrRXHr-2hWdatiFzySYk,5516
18
21
  optimum/rbln/modeling.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
19
22
  optimum/rbln/modeling_alias.py,sha256=pyYGME31QaiBaLqnjQq3LwUF1T6dLeb8QLB91gzNsLM,1574
20
- optimum/rbln/modeling_base.py,sha256=Stmi3ITdw3sJAnSKnWwGin98AjMcKHBK8hviZUejepE,24843
23
+ optimum/rbln/modeling_base.py,sha256=vPkW9yynfa4Ng8XHYw-yKijNpyL3CyiwHnpa6q3KhoY,26545
21
24
  optimum/rbln/modeling_config.py,sha256=R0GBd-upavJrpR-2SvfgCCxP7f5Zr0NxIqdKmwBfVCk,6439
22
- optimum/rbln/modeling_seq2seq.py,sha256=Nps24Sw0DwGsQomzoNq2GOLGygNMNfCr1ZZfycsgiaQ,19004
25
+ optimum/rbln/modeling_seq2seq.py,sha256=eLPiUnei0XGFK01JbIsfS03-hJ_5EkazpoBONv65JNI,19549
23
26
  optimum/rbln/transformers/__init__.py,sha256=RHnckU1ghEE6rEtYrwoElVdKXj3BF2aW0fWRelk65Ng,1860
24
- optimum/rbln/transformers/generation/__init__.py,sha256=334nuQakoo3brXQhHRlrzEGgyhBDx3NPDopyTzuQnVE,1042
25
- optimum/rbln/transformers/generation/streamers.py,sha256=oFsXcOF0c2pjZwP-X-chdtnfJ9vpqGDLEtJHbkHqjUI,5392
27
+ optimum/rbln/transformers/generation/__init__.py,sha256=6MmqS9D21ir4dcH6_fq8kRsX1VK3QspSn6Qw83F4ORE,1081
28
+ optimum/rbln/transformers/generation/streamers.py,sha256=X-dEmL1L_0Oy0QSFj2RNdamb_xbDWLXd-Ms8ckx6OZ4,5778
29
+ optimum/rbln/transformers/generation/utils.py,sha256=F8gnVYG79kzg_IvQynD-p409E_5loy5VaIXvfi094u0,19464
26
30
  optimum/rbln/transformers/models/__init__.py,sha256=LidItK8nZMZi7IoIOq7fyAn4y1M_K6LOSqohVes-dwg,1238
27
31
  optimum/rbln/transformers/models/bart/__init__.py,sha256=SGUcpQ_5iLsVxySxtbwhRpmGt7BgVUTxHAjxAjQStdU,1063
28
32
  optimum/rbln/transformers/models/bart/bart_architecture.py,sha256=H8yVoBFa5uMXQv_wYCHKRW6tIIjdD50ho9C0vcMsbSo,14956
@@ -30,22 +34,22 @@ optimum/rbln/transformers/models/clip/__init__.py,sha256=tbco8qW9QhBe3dtWoKgslLZ
30
34
  optimum/rbln/transformers/models/clip/modeling_clip.py,sha256=SMiMtSWW4EFbhQsHRHKZarTW6u209FTqa0FkM8zN2Tw,4019
31
35
  optimum/rbln/transformers/models/gpt2/__init__.py,sha256=jsOKYXUclG9G6cwUTUX4eeKqjCPfQUwev7TTFIMXS4Y,1040
32
36
  optimum/rbln/transformers/models/gpt2/gpt2_architecture.py,sha256=RQ0Y5f7IKQ73onmuIRB-aw379asSR7kfhUPclLHCOkY,10348
33
- optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=lKJcLrGAPkngKKIYabKgjsE2SWnpxY9IDo7VAOVzxg8,29509
37
+ optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=UbVUmegs-v2jBUQUYwxIQ50EpT2dHDzBiPXshLHtuCQ,15079
34
38
  optimum/rbln/transformers/models/llama/__init__.py,sha256=5mX-MuKzVBj6WQeVxyPhtvFTv0jeZXAFfg4RZ2nVUh0,1042
35
39
  optimum/rbln/transformers/models/llama/llama_architecture.py,sha256=LFaBX8joIHBw5RU-DJnX7soTG4Nfz24mgXS1VHWkTWs,25843
36
- optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=ltpDPDDKLKrEwCilSYCigTIow4ZsUy0oErZYMwi4Z9M,15811
40
+ optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=x8udEqjHKgZQ2QXoyuGTXaT2clfshmI5vds13Kw3I5k,16628
37
41
  optimum/rbln/transformers/models/t5/__init__.py,sha256=dK6F1jbBf001h79WZiVdiNZoXm5kOe2fskzhREhu0EE,1057
38
42
  optimum/rbln/transformers/models/t5/t5_architecture.py,sha256=2nFovfOdiJdY9jdAR9BngwPO3d2Oofn9jqVWgZ-YYZ0,18091
39
43
  optimum/rbln/transformers/models/wav2vec2/__init__.py,sha256=mz4cXqG9b0tDpTAw3qYn3FaJuolX601VmKBE3gohLSw,1043
40
44
  optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=kooQ1CC6p2mHvRHkFE48d69yNTnkG_V6g9Beu6Sy3XU,4063
41
45
  optimum/rbln/transformers/models/whisper/__init__.py,sha256=PZ8qeAAFMas2MizwVYFxlpFWd5k1Pe1x-0IJfYAMhT8,1059
42
- optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=dWXmzQVKM6cQkdyNngLK1j437YjF53sobCWAYIXtyi0,14236
46
+ optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=WV3pcxeXJSw74U4Xc0wW8VFmwdEIdbqrZmcjBQixEN4,14726
43
47
  optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=QX1Nmq26F_82EYgKmdgXEpE2F8ry-inkn2BB9Lx5M38,15885
44
48
  optimum/rbln/utils/__init__.py,sha256=wr7ep1WliFYR0825f7BbHevtT3xhlMSwpDkvAPzShgE,1083
45
49
  optimum/rbln/utils/import_utils.py,sha256=OL3aBy3XLWj7KDb6VZKBPJWiEcktL4qRxlpQpDBcMRg,1116
46
50
  optimum/rbln/utils/runtime_utils.py,sha256=EzEabg2E18nq2WZRDZWsZ_hgrdgQ7u_NElTMAYpSDvM,2545
47
51
  optimum/rbln/utils/save_utils.py,sha256=eFIPtmiblCJ3MvtxEPxmAR3iuLEUrzpyzwtVotDauhw,3283
48
- optimum_rbln-0.1.0.dist-info/METADATA,sha256=jAPVCSO-q6JOIeYLYH2cuolp1LEDwlWOzbIJM88nAgA,6172
49
- optimum_rbln-0.1.0.dist-info/WHEEL,sha256=vnE8JVcI2Wz7GRKorsPArnBdnW2SWKWGow5gu5tHlRU,90
50
- optimum_rbln-0.1.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
51
- optimum_rbln-0.1.0.dist-info/RECORD,,
52
+ optimum_rbln-0.1.1.dist-info/METADATA,sha256=Rfid8XwR-tT2s_Suti1amUKuKJBjIOcOItwteCT2vAg,6172
53
+ optimum_rbln-0.1.1.dist-info/WHEEL,sha256=vnE8JVcI2Wz7GRKorsPArnBdnW2SWKWGow5gu5tHlRU,90
54
+ optimum_rbln-0.1.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
55
+ optimum_rbln-0.1.1.dist-info/RECORD,,