optimum-rbln 0.1.8__py3-none-any.whl → 0.1.9__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 (31) hide show
  1. optimum/rbln/__init__.py +3 -0
  2. optimum/rbln/__version__.py +1 -1
  3. optimum/rbln/diffusers/models/autoencoder_kl.py +3 -3
  4. optimum/rbln/diffusers/models/controlnet.py +4 -3
  5. optimum/rbln/diffusers/models/unet_2d_condition.py +3 -3
  6. optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py +2 -3
  7. optimum/rbln/modeling_alias.py +5 -1
  8. optimum/rbln/modeling_base.py +53 -19
  9. optimum/rbln/transformers/__init__.py +3 -1
  10. optimum/rbln/transformers/models/__init__.py +1 -0
  11. optimum/rbln/transformers/models/clip/modeling_clip.py +1 -1
  12. optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py +4 -3
  13. optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py +137 -22
  14. optimum/rbln/transformers/models/gemma/gemma_architecture.py +10 -3
  15. optimum/rbln/transformers/models/gemma/modeling_gemma.py +9 -3
  16. optimum/rbln/transformers/models/gpt2/modeling_gpt2.py +6 -89
  17. optimum/rbln/transformers/models/llama/modeling_llama.py +9 -3
  18. optimum/rbln/transformers/models/midm/modeling_midm.py +5 -88
  19. optimum/rbln/transformers/models/mistral/__init__.py +24 -0
  20. optimum/rbln/transformers/models/mistral/mistral_architecture.py +29 -0
  21. optimum/rbln/transformers/models/mistral/modeling_mistral.py +68 -0
  22. optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py +1 -1
  23. optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py +8 -2
  24. optimum/rbln/transformers/utils/__init__.py +0 -0
  25. optimum/rbln/transformers/utils/rbln_quantization.py +109 -0
  26. optimum/rbln/utils/import_utils.py +1 -4
  27. optimum/rbln/utils/runtime_utils.py +2 -1
  28. {optimum_rbln-0.1.8.dist-info → optimum_rbln-0.1.9.dist-info}/METADATA +10 -3
  29. {optimum_rbln-0.1.8.dist-info → optimum_rbln-0.1.9.dist-info}/RECORD +31 -26
  30. {optimum_rbln-0.1.8.dist-info → optimum_rbln-0.1.9.dist-info}/WHEEL +0 -0
  31. {optimum_rbln-0.1.8.dist-info → optimum_rbln-0.1.9.dist-info}/licenses/LICENSE +0 -0
@@ -1,13 +1,13 @@
1
- optimum/rbln/__init__.py,sha256=3_h6eOK1OuG84FWaN2zSR7FtuUv4YHuzK4_2DsA6uwk,4785
2
- optimum/rbln/__version__.py,sha256=zemvJ5zjFE6SQT2xmkxc-ZYwNkUTCEX7mz3Epb2qztE,21
1
+ optimum/rbln/__init__.py,sha256=Nci6sT3QWzbyKOuJ_KzbIJd8LqkNLS66TAibBUmnPig,4897
2
+ optimum/rbln/__version__.py,sha256=m1D6fscvvsMhq5HVNKw7kP5M8AqEzQm1ekrn_nLQF1M,21
3
3
  optimum/rbln/diffusers/__init__.py,sha256=w4W7Wy-Mmh8CQZ5M9JnrrE5bN0UsfNehZI41QadE-hk,2605
4
4
  optimum/rbln/diffusers/models/__init__.py,sha256=aY6Llq_31dZjdB9HPBDvi7sXVtdQT9r11gokXG5ffxA,1139
5
- optimum/rbln/diffusers/models/autoencoder_kl.py,sha256=qIhXCfEADNTm2U9I5ZFN1IfA01zwupUY0IBnJwvxLwI,9506
6
- optimum/rbln/diffusers/models/controlnet.py,sha256=BQVjszdk6qlW6HnSXKRsWR7v6oFNcgMG-utNnRPpLSg,9303
7
- optimum/rbln/diffusers/models/unet_2d_condition.py,sha256=oZQyT0PpPcxwmDgCSh7d2nT3oU60iKx-AdgJilzWAb8,14468
5
+ optimum/rbln/diffusers/models/autoencoder_kl.py,sha256=xfjlbbvNmSrxRGlqNmvuCO9wKaRlcpMF7AxZneitTHM,9520
6
+ optimum/rbln/diffusers/models/controlnet.py,sha256=ePIicWNFKwTBjmH5wDsd1C3LipTHpWpE-X5ZGAMQiDU,9329
7
+ optimum/rbln/diffusers/models/unet_2d_condition.py,sha256=zU38gThPlzs7wAPLDdcojMkmx1P25ooQ29TNEQ034YA,14493
8
8
  optimum/rbln/diffusers/pipelines/__init__.py,sha256=Xr_bQbpbC5HbJB2NuUcVQu2BGebDkc2bhsGJmL6jgps,1449
9
9
  optimum/rbln/diffusers/pipelines/controlnet/__init__.py,sha256=k0govvSBxBUR5qpxUGxRMHuQCMX7hXHVZ4EqVRw1LWk,1377
10
- optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py,sha256=FneDaY03zs5aRTeRUTTAbAQsKjGozsea09Km0Z35Ryc,5129
10
+ optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py,sha256=rCvQgf5kiqw_b5pfPTpx2GpjoHW-hQsl_4ikYN9klOc,5128
11
11
  optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet.py,sha256=qX0nDFShyZL3RHKgaeeM8XpMLLDsa_PCrhqS2-IfEwM,42605
12
12
  optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_img2img.py,sha256=9-zw07G5dICByN4k9UgZ5NwPiToRcOApj7M93SM75Pk,41199
13
13
  optimum/rbln/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl.py,sha256=g8SABYroDmcND-0_3CUH1wkdrZPHWdzoWpWLxBk8p-8,53126
@@ -19,55 +19,60 @@ optimum/rbln/diffusers/pipelines/stable_diffusion_xl/__init__.py,sha256=8MDMHIVs
19
19
  optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py,sha256=aCJSoXks7IpjX4rcH6q0TjXtIPzNrbvAvz0KbIEmMr8,5684
20
20
  optimum/rbln/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl_img2img.py,sha256=Cv9L4El2GOE-3IRQKHNEMuSdWGmtVsRnQJShcv2hOo0,5874
21
21
  optimum/rbln/modeling.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
22
- optimum/rbln/modeling_alias.py,sha256=O7vXVl7Qo1tl_UIMMf4z-62pAyKuEZp2x932MTfKoKM,1888
23
- optimum/rbln/modeling_base.py,sha256=U0YbUG8lcKuZwdvY3BLYouvl-etwmXw2BT1Z3etFeeA,34582
22
+ optimum/rbln/modeling_alias.py,sha256=4E0HgaYJTesetqMSnVYKDZ-SRXlTeQ2GZsJ8xuMkmTY,2094
23
+ optimum/rbln/modeling_base.py,sha256=oSm9w1YVCNIQEDW5bdm4bwUF2bqCIGrnrcl335YgVe0,35491
24
24
  optimum/rbln/modeling_config.py,sha256=R0GBd-upavJrpR-2SvfgCCxP7f5Zr0NxIqdKmwBfVCk,6439
25
25
  optimum/rbln/modeling_seq2seq.py,sha256=XXYu_hpxOgQmMgayUy9URQwgMl9Ci2AfWyVHm7tMP5o,16783
26
- optimum/rbln/transformers/__init__.py,sha256=OJX33eQx_iOE864MIRpoYCkstc57UtnNiIaoIATp2Zo,2202
26
+ optimum/rbln/transformers/__init__.py,sha256=ZC7i3PyMbcZDjxE5F93uZRMyrMghGSmG1ZV3rnAr8A4,2269
27
27
  optimum/rbln/transformers/cache_utils.py,sha256=VfArIkAJn3zPXdu-6RXiCWlU-uVwxvhgoMiGxrPtk40,3835
28
28
  optimum/rbln/transformers/generation/__init__.py,sha256=6MmqS9D21ir4dcH6_fq8kRsX1VK3QspSn6Qw83F4ORE,1081
29
29
  optimum/rbln/transformers/generation/streamers.py,sha256=X-dEmL1L_0Oy0QSFj2RNdamb_xbDWLXd-Ms8ckx6OZ4,5778
30
30
  optimum/rbln/transformers/generation/utils.py,sha256=XqPbYQCe6cEuWssG9iHpbbL-wgSQTcyKHabGwKj7lnE,19462
31
- optimum/rbln/transformers/models/__init__.py,sha256=pBJm_uxcyORi9A9UBdHAu3VdFy59PkuYuVek-31RiLs,1404
31
+ optimum/rbln/transformers/models/__init__.py,sha256=kCbAvlMadrhOv8PyEpvJLLBq1b7DYC3zDmlsxLPufTI,1448
32
32
  optimum/rbln/transformers/models/bart/__init__.py,sha256=SGUcpQ_5iLsVxySxtbwhRpmGt7BgVUTxHAjxAjQStdU,1063
33
33
  optimum/rbln/transformers/models/bart/bart_architecture.py,sha256=T9GjcsL8fAJcvAs_ifnZuDP2F77hhbjBcsc7u53k6OE,14951
34
34
  optimum/rbln/transformers/models/clip/__init__.py,sha256=tbco8qW9QhBe3dtWoKgslLZMsXu9dg_KfJ4IgjvK248,1071
35
- optimum/rbln/transformers/models/clip/modeling_clip.py,sha256=WEXc9LWbClEzxeIw-LliI1A5OpiL7nnDm0G2IzYdqp4,3990
35
+ optimum/rbln/transformers/models/clip/modeling_clip.py,sha256=V8Ix1X9gqeBHXP-HvRuNarFRC-cm--ZEU9ICCvwv-tk,4015
36
36
  optimum/rbln/transformers/models/decoderonly/__init__.py,sha256=AG3ib8iZAEDAvVTNhieCyojWZtA67voPB0dI8lbCXTQ,1371
37
- optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=OLN0vvkIlEsP5xQ25x-kk-uw_V201TxWxVMUrGNkW-g,19045
38
- optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=7YfpYWHH3SNCmNYySj7Q-3WSi64pEqMaH1BmDdSYzi8,14191
37
+ optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py,sha256=QJHCrYKAyXOHZsKiRt4uRCxoB7zATxjOERSbqOqjK-g,19095
38
+ optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py,sha256=DpY88CqhhesQVh6z42nvzG9p1Ju9wuEloFjPT8F92t4,18436
39
39
  optimum/rbln/transformers/models/dpt/__init__.py,sha256=R8OHDxOAYPjkk5t8osaPqRh85Pf1Cg1BtzqesqFRcTI,1045
40
40
  optimum/rbln/transformers/models/dpt/modeling_dpt.py,sha256=uN_5DhjGbFmTbpm1JUBgPsDhMP_vIyc0QM2UK5DoRqc,3537
41
41
  optimum/rbln/transformers/models/gemma/__init__.py,sha256=L1Qfr6dufWtoUZND_ofwxXPSxivEvPR8exy16a_nM2o,1042
42
- optimum/rbln/transformers/models/gemma/gemma_architecture.py,sha256=nSLyuw9xMs2f17fVMNQiiuwRp813HP9q2uIyFhnDZ3o,4291
43
- optimum/rbln/transformers/models/gemma/modeling_gemma.py,sha256=d47ahENZOJ2pNxyZr1ywvnUQhdBjMk5o72rjcqqLz0M,2465
42
+ optimum/rbln/transformers/models/gemma/gemma_architecture.py,sha256=hT0CqL_jhKWi8cDa1zFcAFPyli844wkliJ3bL5OyEdQ,4376
43
+ optimum/rbln/transformers/models/gemma/modeling_gemma.py,sha256=QW57x2ehojt0CHBJSC4Y51VaMqvjbiSYfEpxIPVMbj8,2643
44
44
  optimum/rbln/transformers/models/gpt2/__init__.py,sha256=jsOKYXUclG9G6cwUTUX4eeKqjCPfQUwev7TTFIMXS4Y,1040
45
45
  optimum/rbln/transformers/models/gpt2/gpt2_architecture.py,sha256=SIzwfQFRNvNOHjZmKcUZiURbnmXNXFh95twYELVmHcg,10278
46
- optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=TWfis7dqzPbm23cHfzLl43TwLzyS5nZMcgxSqxBnD1w,5513
46
+ optimum/rbln/transformers/models/gpt2/modeling_gpt2.py,sha256=ZR0wep7V_sENaODlAbg70_xNZKeTK8xdN0AEOsIrKms,2721
47
47
  optimum/rbln/transformers/models/llama/__init__.py,sha256=5mX-MuKzVBj6WQeVxyPhtvFTv0jeZXAFfg4RZ2nVUh0,1042
48
48
  optimum/rbln/transformers/models/llama/llama_architecture.py,sha256=j4mifSOaIk7wwV9fL9wQSt5kR3rpnvjtxd3VzhMNdgY,1123
49
- optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=RXv7yyf3dC1h02seAxpRlzrvofgdy9OWE4P8SRwUlk0,2465
49
+ optimum/rbln/transformers/models/llama/modeling_llama.py,sha256=AQU4RVWQb0Ht_eAEiNTkcEq9bPCr-P1JnCkLy77yDnE,2643
50
50
  optimum/rbln/transformers/models/midm/__init__.py,sha256=_6kYchy47frGMZ8uoUspZ9IwrmCBQJ-8kVfXM7xOMew,1249
51
51
  optimum/rbln/transformers/models/midm/hf_hub_cached/configuration_midm.py,sha256=P5JqTTcx56HOccxKbR14ZjA67BI0RNnJycG738JMaJ4,833
52
52
  optimum/rbln/transformers/models/midm/hf_hub_cached/midm_bitext_tokenization.py,sha256=5lhMXfqnIak1PJ9YL-vUxIdY_3DUr3IBXzTqf3ofpmI,12835
53
53
  optimum/rbln/transformers/models/midm/hf_hub_cached/modeling_midm.py,sha256=54__wd9EXwGxmaHDksTTcUD2aWl6WoszYsR8dlL1wfE,61031
54
54
  optimum/rbln/transformers/models/midm/hf_hub_cached/rotary_position_embedding.py,sha256=5ywaUVKTvqO8GRsHOSXOOGlbiEn-DbGkpJs59_dFb18,4059
55
55
  optimum/rbln/transformers/models/midm/midm_architecture.py,sha256=IFnu54MVPFEk5pvaeJ8RZGlCR7X-9MMeTKThXOP6_M0,11367
56
- optimum/rbln/transformers/models/midm/modeling_midm.py,sha256=y16ALbYdvlpAhXxrXvVmDYk6uc27u79J8SognYVC-VE,5504
56
+ optimum/rbln/transformers/models/midm/modeling_midm.py,sha256=rk1LgR3s9dDC1zwE6Jg6LJQK868VyAIViD9zsK09-UE,2779
57
+ optimum/rbln/transformers/models/mistral/__init__.py,sha256=XtuOmzBITjj-H1yctXobJjHF908x1Wlxr_p4hi06v8I,1046
58
+ optimum/rbln/transformers/models/mistral/mistral_architecture.py,sha256=LCvY4L0Wq1VruKhZ3JTSiuZJqQRJlTae5A2bKsUBGAg,1128
59
+ optimum/rbln/transformers/models/mistral/modeling_mistral.py,sha256=77CRdlD3n465fQRZ6SOya9jRgL7M3KN624VzRUJBxt4,2678
57
60
  optimum/rbln/transformers/models/t5/__init__.py,sha256=dK6F1jbBf001h79WZiVdiNZoXm5kOe2fskzhREhu0EE,1057
58
61
  optimum/rbln/transformers/models/t5/t5_architecture.py,sha256=2nFovfOdiJdY9jdAR9BngwPO3d2Oofn9jqVWgZ-YYZ0,18091
59
62
  optimum/rbln/transformers/models/wav2vec2/__init__.py,sha256=mz4cXqG9b0tDpTAw3qYn3FaJuolX601VmKBE3gohLSw,1043
60
- optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=kooQ1CC6p2mHvRHkFE48d69yNTnkG_V6g9Beu6Sy3XU,4063
63
+ optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py,sha256=ZnsJ9f2Lu3WNH2QFiuOCTU5Y--_wNlyYy_0zuvkJ5RI,4088
61
64
  optimum/rbln/transformers/models/whisper/__init__.py,sha256=PZ8qeAAFMas2MizwVYFxlpFWd5k1Pe1x-0IJfYAMhT8,1059
62
65
  optimum/rbln/transformers/models/whisper/modeling_whisper.py,sha256=L49ThCv5sqidNevBGsCpGrOSH4H6wzXOCmON1PCmY9M,11996
63
66
  optimum/rbln/transformers/models/whisper/whisper_architecture.py,sha256=QtHP5bXUa9HXa95rEdaqsmbH3eG2QJgUBTV1OG7LF-I,15879
64
67
  optimum/rbln/transformers/models/xlm_roberta/__init__.py,sha256=NTj4hCpd8L2_i5DZuV5wp-h8OlTLYVUqTrJxzY_Dg9g,1047
65
- optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py,sha256=krLNMjsmK2I_Ai2JUJzLqE1HIrTDNhpTY2EDQFQV6Kg,4971
68
+ optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py,sha256=YOnybpasUwtAhZSRirHQj0kvmzpD0i2nBzcYT0En3ew,5018
69
+ optimum/rbln/transformers/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
70
+ optimum/rbln/transformers/utils/rbln_quantization.py,sha256=kRms512Vf1o7vTZl5cd64ivjBVltZrUDCVqCRWFft88,3734
66
71
  optimum/rbln/utils/__init__.py,sha256=F6hJP00eV1_hT_IVwqqYwLWcLQAvZbmmrNMJTia3mjI,1106
67
- optimum/rbln/utils/import_utils.py,sha256=i2GmQJC9kl4BvXncVUrqx8VCqfv1omaHiWyCliBxChg,2632
68
- optimum/rbln/utils/runtime_utils.py,sha256=EzEabg2E18nq2WZRDZWsZ_hgrdgQ7u_NElTMAYpSDvM,2545
72
+ optimum/rbln/utils/import_utils.py,sha256=btQpDE5WBJUfCMzg5S-cokWbTh_IHYGo1tRNiU16vLU,2624
73
+ optimum/rbln/utils/runtime_utils.py,sha256=6APwOmW04DjdRto5ntKZFTw4CuFd194OcQtImcIQD2U,2621
69
74
  optimum/rbln/utils/save_utils.py,sha256=eFIPtmiblCJ3MvtxEPxmAR3iuLEUrzpyzwtVotDauhw,3283
70
- optimum_rbln-0.1.8.dist-info/METADATA,sha256=UJKCmPXw-b8Jm1ie2bAWVowLdYQWUuYD-Tz3-WGwksk,4311
71
- optimum_rbln-0.1.8.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
72
- optimum_rbln-0.1.8.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
73
- optimum_rbln-0.1.8.dist-info/RECORD,,
75
+ optimum_rbln-0.1.9.dist-info/METADATA,sha256=loqigjmaw3tu7yv9gzydFI-JMJcdr9TTGuJWMQx5F44,4578
76
+ optimum_rbln-0.1.9.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
77
+ optimum_rbln-0.1.9.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
78
+ optimum_rbln-0.1.9.dist-info/RECORD,,