optimum-rbln 0.7.4a4__py3-none-any.whl → 0.7.4a6__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 (101) hide show
  1. optimum/rbln/__init__.py +164 -36
  2. optimum/rbln/__version__.py +2 -2
  3. optimum/rbln/configuration_utils.py +772 -0
  4. optimum/rbln/diffusers/__init__.py +56 -0
  5. optimum/rbln/diffusers/configurations/__init__.py +30 -0
  6. optimum/rbln/diffusers/configurations/models/__init__.py +6 -0
  7. optimum/rbln/diffusers/configurations/models/configuration_autoencoder_kl.py +66 -0
  8. optimum/rbln/diffusers/configurations/models/configuration_controlnet.py +54 -0
  9. optimum/rbln/diffusers/configurations/models/configuration_prior_transformer.py +44 -0
  10. optimum/rbln/diffusers/configurations/models/configuration_transformer_sd3.py +48 -0
  11. optimum/rbln/diffusers/configurations/models/configuration_unet_2d_condition.py +66 -0
  12. optimum/rbln/diffusers/configurations/models/configuration_vq_model.py +67 -0
  13. optimum/rbln/diffusers/configurations/pipelines/__init__.py +30 -0
  14. optimum/rbln/diffusers/configurations/pipelines/configuration_controlnet.py +221 -0
  15. optimum/rbln/diffusers/configurations/pipelines/configuration_kandinsky2_2.py +285 -0
  16. optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion.py +118 -0
  17. optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion_3.py +143 -0
  18. optimum/rbln/diffusers/configurations/pipelines/configuration_stable_diffusion_xl.py +124 -0
  19. optimum/rbln/diffusers/modeling_diffusers.py +63 -122
  20. optimum/rbln/diffusers/models/autoencoders/autoencoder_kl.py +109 -128
  21. optimum/rbln/diffusers/models/autoencoders/vae.py +4 -6
  22. optimum/rbln/diffusers/models/autoencoders/vq_model.py +84 -85
  23. optimum/rbln/diffusers/models/controlnet.py +55 -70
  24. optimum/rbln/diffusers/models/transformers/prior_transformer.py +40 -77
  25. optimum/rbln/diffusers/models/transformers/transformer_sd3.py +43 -68
  26. optimum/rbln/diffusers/models/unets/unet_2d_condition.py +110 -113
  27. optimum/rbln/diffusers/pipelines/controlnet/multicontrolnet.py +3 -4
  28. optimum/rbln/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +2 -1
  29. optimum/rbln/modeling.py +58 -39
  30. optimum/rbln/modeling_base.py +107 -78
  31. optimum/rbln/transformers/__init__.py +87 -8
  32. optimum/rbln/transformers/configuration_alias.py +49 -0
  33. optimum/rbln/transformers/configuration_generic.py +142 -0
  34. optimum/rbln/transformers/modeling_generic.py +193 -280
  35. optimum/rbln/transformers/models/__init__.py +108 -34
  36. optimum/rbln/transformers/models/auto/auto_factory.py +3 -3
  37. optimum/rbln/transformers/models/bart/__init__.py +1 -0
  38. optimum/rbln/transformers/models/bart/configuration_bart.py +24 -0
  39. optimum/rbln/transformers/models/bart/modeling_bart.py +10 -84
  40. optimum/rbln/transformers/models/bert/__init__.py +1 -0
  41. optimum/rbln/transformers/models/bert/configuration_bert.py +31 -0
  42. optimum/rbln/transformers/models/bert/modeling_bert.py +7 -80
  43. optimum/rbln/transformers/models/clip/__init__.py +6 -0
  44. optimum/rbln/transformers/models/clip/configuration_clip.py +79 -0
  45. optimum/rbln/transformers/models/clip/modeling_clip.py +72 -75
  46. optimum/rbln/transformers/models/decoderonly/__init__.py +1 -0
  47. optimum/rbln/transformers/models/decoderonly/configuration_decoderonly.py +90 -0
  48. optimum/rbln/transformers/models/decoderonly/decoderonly_architecture.py +115 -84
  49. optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py +282 -216
  50. optimum/rbln/transformers/models/dpt/__init__.py +1 -0
  51. optimum/rbln/transformers/models/dpt/configuration_dpt.py +19 -0
  52. optimum/rbln/transformers/models/dpt/modeling_dpt.py +3 -76
  53. optimum/rbln/transformers/models/exaone/__init__.py +1 -0
  54. optimum/rbln/transformers/models/exaone/configuration_exaone.py +19 -0
  55. optimum/rbln/transformers/models/gemma/__init__.py +1 -0
  56. optimum/rbln/transformers/models/gemma/configuration_gemma.py +19 -0
  57. optimum/rbln/transformers/models/gpt2/__init__.py +1 -0
  58. optimum/rbln/transformers/models/gpt2/configuration_gpt2.py +19 -0
  59. optimum/rbln/transformers/models/llama/__init__.py +1 -0
  60. optimum/rbln/transformers/models/llama/configuration_llama.py +19 -0
  61. optimum/rbln/transformers/models/llava_next/__init__.py +1 -0
  62. optimum/rbln/transformers/models/llava_next/configuration_llava_next.py +46 -0
  63. optimum/rbln/transformers/models/llava_next/modeling_llava_next.py +12 -23
  64. optimum/rbln/transformers/models/midm/__init__.py +1 -0
  65. optimum/rbln/transformers/models/midm/configuration_midm.py +19 -0
  66. optimum/rbln/transformers/models/mistral/__init__.py +1 -0
  67. optimum/rbln/transformers/models/mistral/configuration_mistral.py +19 -0
  68. optimum/rbln/transformers/models/phi/__init__.py +1 -0
  69. optimum/rbln/transformers/models/phi/configuration_phi.py +19 -0
  70. optimum/rbln/transformers/models/qwen2/__init__.py +1 -0
  71. optimum/rbln/transformers/models/qwen2/configuration_qwen2.py +19 -0
  72. optimum/rbln/transformers/models/qwen2_5_vl/__init__.py +19 -0
  73. optimum/rbln/transformers/models/qwen2_5_vl/configuration_qwen2_5_vl.py +68 -0
  74. optimum/rbln/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py +608 -0
  75. optimum/rbln/transformers/models/qwen2_5_vl/qwen2_5_vl_architecture.py +214 -0
  76. optimum/rbln/transformers/models/seq2seq/__init__.py +1 -0
  77. optimum/rbln/transformers/models/seq2seq/configuration_seq2seq2.py +66 -0
  78. optimum/rbln/transformers/models/seq2seq/modeling_seq2seq.py +80 -97
  79. optimum/rbln/transformers/models/t5/__init__.py +1 -0
  80. optimum/rbln/transformers/models/t5/configuration_t5.py +24 -0
  81. optimum/rbln/transformers/models/t5/modeling_t5.py +22 -150
  82. optimum/rbln/transformers/models/time_series_transformers/__init__.py +1 -0
  83. optimum/rbln/transformers/models/time_series_transformers/configuration_time_series_transformer.py +34 -0
  84. optimum/rbln/transformers/models/time_series_transformers/modeling_time_series_transformers.py +52 -54
  85. optimum/rbln/transformers/models/wav2vec2/__init__.py +1 -0
  86. optimum/rbln/transformers/models/wav2vec2/configuration_wav2vec.py +19 -0
  87. optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py +9 -72
  88. optimum/rbln/transformers/models/whisper/__init__.py +1 -0
  89. optimum/rbln/transformers/models/whisper/configuration_whisper.py +64 -0
  90. optimum/rbln/transformers/models/whisper/modeling_whisper.py +57 -72
  91. optimum/rbln/transformers/models/xlm_roberta/__init__.py +1 -0
  92. optimum/rbln/transformers/models/xlm_roberta/configuration_xlm_roberta.py +19 -0
  93. optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py +3 -83
  94. optimum/rbln/utils/runtime_utils.py +33 -2
  95. optimum/rbln/utils/submodule.py +26 -43
  96. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a6.dist-info}/METADATA +1 -1
  97. optimum_rbln-0.7.4a6.dist-info/RECORD +166 -0
  98. optimum/rbln/modeling_config.py +0 -310
  99. optimum_rbln-0.7.4a4.dist-info/RECORD +0 -126
  100. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a6.dist-info}/WHEEL +0 -0
  101. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a6.dist-info}/licenses/LICENSE +0 -0
optimum/rbln/__init__.py CHANGED
@@ -25,11 +25,13 @@ _import_structure = {
25
25
  "RBLNBaseModel",
26
26
  "RBLNModel",
27
27
  ],
28
- "modeling_config": [
28
+ "configuration_utils": [
29
29
  "RBLNCompileConfig",
30
- "RBLNConfig",
30
+ "RBLNModelConfig",
31
31
  ],
32
32
  "transformers": [
33
+ "RBLNASTForAudioClassification",
34
+ "RBLNASTForAudioClassificationConfig",
33
35
  "RBLNAutoModel",
34
36
  "RBLNAutoModelForAudioClassification",
35
37
  "RBLNAutoModelForCausalLM",
@@ -43,111 +45,200 @@ _import_structure = {
43
45
  "RBLNAutoModelForSpeechSeq2Seq",
44
46
  "RBLNAutoModelForVision2Seq",
45
47
  "RBLNBartForConditionalGeneration",
48
+ "RBLNBartForConditionalGenerationConfig",
46
49
  "RBLNBartModel",
50
+ "RBLNBartModelConfig",
51
+ "RBLNBertForMaskedLM",
52
+ "RBLNBertForMaskedLMConfig",
53
+ "RBLNBertForQuestionAnswering",
54
+ "RBLNBertForQuestionAnsweringConfig",
47
55
  "RBLNBertModel",
56
+ "RBLNBertModelConfig",
48
57
  "RBLNCLIPTextModel",
58
+ "RBLNCLIPTextModelConfig",
49
59
  "RBLNCLIPTextModelWithProjection",
60
+ "RBLNCLIPTextModelWithProjectionConfig",
50
61
  "RBLNCLIPVisionModel",
62
+ "RBLNCLIPVisionModelConfig",
51
63
  "RBLNCLIPVisionModelWithProjection",
64
+ "RBLNCLIPVisionModelWithProjectionConfig",
65
+ "RBLNDecoderOnlyModelForCausalLM",
66
+ "RBLNDecoderOnlyModelForCausalLMConfig",
67
+ "RBLNDistilBertForQuestionAnswering",
68
+ "RBLNDistilBertForQuestionAnsweringConfig",
52
69
  "RBLNDPTForDepthEstimation",
70
+ "RBLNDPTForDepthEstimationConfig",
53
71
  "RBLNExaoneForCausalLM",
72
+ "RBLNExaoneForCausalLMConfig",
54
73
  "RBLNGemmaForCausalLM",
74
+ "RBLNGemmaForCausalLMConfig",
55
75
  "RBLNGPT2LMHeadModel",
56
- "RBLNQwen2ForCausalLM",
57
- "RBLNWav2Vec2ForCTC",
76
+ "RBLNGPT2LMHeadModelConfig",
58
77
  "RBLNLlamaForCausalLM",
59
- "RBLNT5EncoderModel",
60
- "RBLNT5ForConditionalGeneration",
61
- "RBLNPhiForCausalLM",
78
+ "RBLNLlamaForCausalLMConfig",
62
79
  "RBLNLlavaNextForConditionalGeneration",
80
+ "RBLNLlavaNextForConditionalGenerationConfig",
63
81
  "RBLNMidmLMHeadModel",
82
+ "RBLNMidmLMHeadModelConfig",
64
83
  "RBLNMistralForCausalLM",
65
- "RBLNWhisperForConditionalGeneration",
66
- "RBLNXLMRobertaModel",
67
- "RBLNASTForAudioClassification",
68
- "RBLNBertForQuestionAnswering",
69
- "RBLNDistilBertForQuestionAnswering",
84
+ "RBLNMistralForCausalLMConfig",
85
+ "RBLNPhiForCausalLM",
86
+ "RBLNPhiForCausalLMConfig",
87
+ "RBLNQwen2ForCausalLM",
88
+ "RBLNQwen2ForCausalLMConfig",
89
+ "RBLNQwen2_5_VisionTransformerPretrainedModel",
90
+ "RBLNQwen2_5_VisionTransformerPretrainedModelConfig",
91
+ "RBLNQwen2_5_VLForConditionalGeneration",
92
+ "RBLNQwen2_5_VLForConditionalGenerationConfig",
70
93
  "RBLNResNetForImageClassification",
71
- "RBLNXLMRobertaForSequenceClassification",
72
- "RBLNRobertaForSequenceClassification",
94
+ "RBLNResNetForImageClassificationConfig",
73
95
  "RBLNRobertaForMaskedLM",
74
- "RBLNViTForImageClassification",
75
- "RBLNBertForMaskedLM",
96
+ "RBLNRobertaForMaskedLMConfig",
97
+ "RBLNRobertaForSequenceClassification",
98
+ "RBLNRobertaForSequenceClassificationConfig",
99
+ "RBLNT5EncoderModel",
100
+ "RBLNT5EncoderModelConfig",
101
+ "RBLNT5ForConditionalGeneration",
102
+ "RBLNT5ForConditionalGenerationConfig",
76
103
  "RBLNTimeSeriesTransformerForPrediction",
104
+ "RBLNTimeSeriesTransformerForPredictionConfig",
105
+ "RBLNViTForImageClassification",
106
+ "RBLNViTForImageClassificationConfig",
107
+ "RBLNWav2Vec2ForCTC",
108
+ "RBLNWav2Vec2ForCTCConfig",
109
+ "RBLNWhisperForConditionalGeneration",
110
+ "RBLNWhisperForConditionalGenerationConfig",
111
+ "RBLNXLMRobertaForSequenceClassification",
112
+ "RBLNXLMRobertaForSequenceClassificationConfig",
113
+ "RBLNXLMRobertaModel",
114
+ "RBLNXLMRobertaModelConfig",
77
115
  ],
78
116
  "diffusers": [
79
117
  "RBLNAutoencoderKL",
118
+ "RBLNAutoencoderKLConfig",
80
119
  "RBLNControlNetModel",
81
- "RBLNPriorTransformer",
120
+ "RBLNControlNetModelConfig",
121
+ "RBLNDiffusionMixin",
82
122
  "RBLNKandinskyV22CombinedPipeline",
123
+ "RBLNKandinskyV22CombinedPipelineConfig",
83
124
  "RBLNKandinskyV22Img2ImgCombinedPipeline",
125
+ "RBLNKandinskyV22Img2ImgCombinedPipelineConfig",
126
+ "RBLNKandinskyV22Img2ImgPipeline",
127
+ "RBLNKandinskyV22Img2ImgPipelineConfig",
84
128
  "RBLNKandinskyV22InpaintCombinedPipeline",
129
+ "RBLNKandinskyV22InpaintCombinedPipelineConfig",
85
130
  "RBLNKandinskyV22InpaintPipeline",
86
- "RBLNKandinskyV22Img2ImgPipeline",
87
- "RBLNKandinskyV22PriorPipeline",
131
+ "RBLNKandinskyV22InpaintPipelineConfig",
88
132
  "RBLNKandinskyV22Pipeline",
89
- "RBLNStableDiffusionPipeline",
90
- "RBLNStableDiffusionXLPipeline",
91
- "RBLNUNet2DConditionModel",
92
- "RBLNStableDiffusionImg2ImgPipeline",
93
- "RBLNStableDiffusionInpaintPipeline",
94
- "RBLNStableDiffusionControlNetImg2ImgPipeline",
133
+ "RBLNKandinskyV22PipelineConfig",
134
+ "RBLNKandinskyV22PriorPipeline",
135
+ "RBLNKandinskyV22PriorPipelineConfig",
95
136
  "RBLNMultiControlNetModel",
96
- "RBLNStableDiffusionXLImg2ImgPipeline",
97
- "RBLNStableDiffusionXLInpaintPipeline",
98
- "RBLNStableDiffusionControlNetPipeline",
99
- "RBLNStableDiffusionXLControlNetPipeline",
100
- "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
101
- "RBLNVQModel",
137
+ "RBLNPriorTransformer",
138
+ "RBLNPriorTransformerConfig",
102
139
  "RBLNSD3Transformer2DModel",
140
+ "RBLNSD3Transformer2DModelConfig",
103
141
  "RBLNStableDiffusion3Img2ImgPipeline",
142
+ "RBLNStableDiffusion3Img2ImgPipelineConfig",
104
143
  "RBLNStableDiffusion3InpaintPipeline",
144
+ "RBLNStableDiffusion3InpaintPipelineConfig",
105
145
  "RBLNStableDiffusion3Pipeline",
106
- "RBLNDiffusionMixin",
146
+ "RBLNStableDiffusion3PipelineConfig",
147
+ "RBLNStableDiffusionControlNetImg2ImgPipeline",
148
+ "RBLNStableDiffusionControlNetImg2ImgPipelineConfig",
149
+ "RBLNStableDiffusionControlNetPipeline",
150
+ "RBLNStableDiffusionControlNetPipelineConfig",
151
+ "RBLNStableDiffusionImg2ImgPipeline",
152
+ "RBLNStableDiffusionImg2ImgPipelineConfig",
153
+ "RBLNStableDiffusionInpaintPipeline",
154
+ "RBLNStableDiffusionInpaintPipelineConfig",
155
+ "RBLNStableDiffusionPipeline",
156
+ "RBLNStableDiffusionPipelineConfig",
157
+ "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
158
+ "RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig",
159
+ "RBLNStableDiffusionXLControlNetPipeline",
160
+ "RBLNStableDiffusionXLControlNetPipelineConfig",
161
+ "RBLNStableDiffusionXLImg2ImgPipeline",
162
+ "RBLNStableDiffusionXLImg2ImgPipelineConfig",
163
+ "RBLNStableDiffusionXLInpaintPipeline",
164
+ "RBLNStableDiffusionXLInpaintPipelineConfig",
165
+ "RBLNStableDiffusionXLPipeline",
166
+ "RBLNStableDiffusionXLPipelineConfig",
167
+ "RBLNUNet2DConditionModel",
168
+ "RBLNUNet2DConditionModelConfig",
169
+ "RBLNVQModel",
170
+ "RBLNVQModelConfig",
107
171
  ],
108
172
  }
109
173
 
110
174
  if TYPE_CHECKING:
175
+ from .configuration_utils import (
176
+ RBLNCompileConfig,
177
+ RBLNModelConfig,
178
+ )
111
179
  from .diffusers import (
112
180
  RBLNAutoencoderKL,
181
+ RBLNAutoencoderKLConfig,
113
182
  RBLNControlNetModel,
183
+ RBLNControlNetModelConfig,
114
184
  RBLNDiffusionMixin,
115
185
  RBLNKandinskyV22CombinedPipeline,
186
+ RBLNKandinskyV22CombinedPipelineConfig,
116
187
  RBLNKandinskyV22Img2ImgCombinedPipeline,
188
+ RBLNKandinskyV22Img2ImgCombinedPipelineConfig,
117
189
  RBLNKandinskyV22Img2ImgPipeline,
190
+ RBLNKandinskyV22Img2ImgPipelineConfig,
118
191
  RBLNKandinskyV22InpaintCombinedPipeline,
192
+ RBLNKandinskyV22InpaintCombinedPipelineConfig,
119
193
  RBLNKandinskyV22InpaintPipeline,
194
+ RBLNKandinskyV22InpaintPipelineConfig,
120
195
  RBLNKandinskyV22Pipeline,
196
+ RBLNKandinskyV22PipelineConfig,
121
197
  RBLNKandinskyV22PriorPipeline,
198
+ RBLNKandinskyV22PriorPipelineConfig,
122
199
  RBLNMultiControlNetModel,
123
200
  RBLNPriorTransformer,
201
+ RBLNPriorTransformerConfig,
124
202
  RBLNSD3Transformer2DModel,
203
+ RBLNSD3Transformer2DModelConfig,
125
204
  RBLNStableDiffusion3Img2ImgPipeline,
205
+ RBLNStableDiffusion3Img2ImgPipelineConfig,
126
206
  RBLNStableDiffusion3InpaintPipeline,
207
+ RBLNStableDiffusion3InpaintPipelineConfig,
127
208
  RBLNStableDiffusion3Pipeline,
209
+ RBLNStableDiffusion3PipelineConfig,
128
210
  RBLNStableDiffusionControlNetImg2ImgPipeline,
211
+ RBLNStableDiffusionControlNetImg2ImgPipelineConfig,
129
212
  RBLNStableDiffusionControlNetPipeline,
213
+ RBLNStableDiffusionControlNetPipelineConfig,
130
214
  RBLNStableDiffusionImg2ImgPipeline,
215
+ RBLNStableDiffusionImg2ImgPipelineConfig,
131
216
  RBLNStableDiffusionInpaintPipeline,
217
+ RBLNStableDiffusionInpaintPipelineConfig,
132
218
  RBLNStableDiffusionPipeline,
219
+ RBLNStableDiffusionPipelineConfig,
133
220
  RBLNStableDiffusionXLControlNetImg2ImgPipeline,
221
+ RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig,
134
222
  RBLNStableDiffusionXLControlNetPipeline,
223
+ RBLNStableDiffusionXLControlNetPipelineConfig,
135
224
  RBLNStableDiffusionXLImg2ImgPipeline,
225
+ RBLNStableDiffusionXLImg2ImgPipelineConfig,
136
226
  RBLNStableDiffusionXLInpaintPipeline,
227
+ RBLNStableDiffusionXLInpaintPipelineConfig,
137
228
  RBLNStableDiffusionXLPipeline,
229
+ RBLNStableDiffusionXLPipelineConfig,
138
230
  RBLNUNet2DConditionModel,
231
+ RBLNUNet2DConditionModelConfig,
139
232
  RBLNVQModel,
233
+ RBLNVQModelConfig,
140
234
  )
141
235
  from .modeling import (
142
236
  RBLNBaseModel,
143
237
  RBLNModel,
144
238
  )
145
- from .modeling_config import (
146
- RBLNCompileConfig,
147
- RBLNConfig,
148
- )
149
239
  from .transformers import (
150
240
  RBLNASTForAudioClassification,
241
+ RBLNASTForAudioClassificationConfig,
151
242
  RBLNAutoModel,
152
243
  RBLNAutoModelForAudioClassification,
153
244
  RBLNAutoModelForCausalLM,
@@ -161,36 +252,73 @@ if TYPE_CHECKING:
161
252
  RBLNAutoModelForSpeechSeq2Seq,
162
253
  RBLNAutoModelForVision2Seq,
163
254
  RBLNBartForConditionalGeneration,
255
+ RBLNBartForConditionalGenerationConfig,
164
256
  RBLNBartModel,
257
+ RBLNBartModelConfig,
165
258
  RBLNBertForMaskedLM,
259
+ RBLNBertForMaskedLMConfig,
166
260
  RBLNBertForQuestionAnswering,
261
+ RBLNBertForQuestionAnsweringConfig,
167
262
  RBLNBertModel,
263
+ RBLNBertModelConfig,
168
264
  RBLNCLIPTextModel,
265
+ RBLNCLIPTextModelConfig,
169
266
  RBLNCLIPTextModelWithProjection,
267
+ RBLNCLIPTextModelWithProjectionConfig,
170
268
  RBLNCLIPVisionModel,
269
+ RBLNCLIPVisionModelConfig,
171
270
  RBLNCLIPVisionModelWithProjection,
271
+ RBLNCLIPVisionModelWithProjectionConfig,
272
+ RBLNDecoderOnlyModelForCausalLM,
273
+ RBLNDecoderOnlyModelForCausalLMConfig,
172
274
  RBLNDistilBertForQuestionAnswering,
275
+ RBLNDistilBertForQuestionAnsweringConfig,
173
276
  RBLNDPTForDepthEstimation,
277
+ RBLNDPTForDepthEstimationConfig,
174
278
  RBLNExaoneForCausalLM,
279
+ RBLNExaoneForCausalLMConfig,
175
280
  RBLNGemmaForCausalLM,
281
+ RBLNGemmaForCausalLMConfig,
176
282
  RBLNGPT2LMHeadModel,
283
+ RBLNGPT2LMHeadModelConfig,
177
284
  RBLNLlamaForCausalLM,
285
+ RBLNLlamaForCausalLMConfig,
178
286
  RBLNLlavaNextForConditionalGeneration,
287
+ RBLNLlavaNextForConditionalGenerationConfig,
179
288
  RBLNMidmLMHeadModel,
289
+ RBLNMidmLMHeadModelConfig,
180
290
  RBLNMistralForCausalLM,
291
+ RBLNMistralForCausalLMConfig,
181
292
  RBLNPhiForCausalLM,
293
+ RBLNPhiForCausalLMConfig,
294
+ RBLNQwen2_5_VisionTransformerPretrainedModel,
295
+ RBLNQwen2_5_VisionTransformerPretrainedModelConfig,
296
+ RBLNQwen2_5_VLForConditionalGeneration,
297
+ RBLNQwen2_5_VLForConditionalGenerationConfig,
182
298
  RBLNQwen2ForCausalLM,
299
+ RBLNQwen2ForCausalLMConfig,
183
300
  RBLNResNetForImageClassification,
301
+ RBLNResNetForImageClassificationConfig,
184
302
  RBLNRobertaForMaskedLM,
303
+ RBLNRobertaForMaskedLMConfig,
185
304
  RBLNRobertaForSequenceClassification,
305
+ RBLNRobertaForSequenceClassificationConfig,
186
306
  RBLNT5EncoderModel,
307
+ RBLNT5EncoderModelConfig,
187
308
  RBLNT5ForConditionalGeneration,
309
+ RBLNT5ForConditionalGenerationConfig,
188
310
  RBLNTimeSeriesTransformerForPrediction,
311
+ RBLNTimeSeriesTransformerForPredictionConfig,
189
312
  RBLNViTForImageClassification,
313
+ RBLNViTForImageClassificationConfig,
190
314
  RBLNWav2Vec2ForCTC,
315
+ RBLNWav2Vec2ForCTCConfig,
191
316
  RBLNWhisperForConditionalGeneration,
317
+ RBLNWhisperForConditionalGenerationConfig,
192
318
  RBLNXLMRobertaForSequenceClassification,
319
+ RBLNXLMRobertaForSequenceClassificationConfig,
193
320
  RBLNXLMRobertaModel,
321
+ RBLNXLMRobertaModelConfig,
194
322
  )
195
323
 
196
324
  else:
optimum/rbln/__version__.py CHANGED
@@ -17,5 +17,5 @@ __version__: str
17
17
  __version_tuple__: VERSION_TUPLE
18
18
  version_tuple: VERSION_TUPLE
19
19
 
20
- __version__ = version = '0.7.4a4'
21
- __version_tuple__ = version_tuple = (0, 7, 4)
20
+ __version__ = version = '0.7.4a6'
21
+ __version_tuple__ = version_tuple = (0, 7, 4, 'a6')