optimum-rbln 0.7.4a4__py3-none-any.whl → 0.7.4a5__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 (96) hide show
  1. optimum/rbln/__init__.py +156 -36
  2. optimum/rbln/__version__.py +1 -1
  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 +85 -75
  31. optimum/rbln/transformers/__init__.py +79 -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 +96 -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 +50 -43
  49. optimum/rbln/transformers/models/decoderonly/modeling_decoderonly.py +114 -141
  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/seq2seq/__init__.py +1 -0
  73. optimum/rbln/transformers/models/seq2seq/configuration_seq2seq2.py +66 -0
  74. optimum/rbln/transformers/models/seq2seq/modeling_seq2seq.py +80 -97
  75. optimum/rbln/transformers/models/t5/__init__.py +1 -0
  76. optimum/rbln/transformers/models/t5/configuration_t5.py +24 -0
  77. optimum/rbln/transformers/models/t5/modeling_t5.py +22 -150
  78. optimum/rbln/transformers/models/time_series_transformers/__init__.py +1 -0
  79. optimum/rbln/transformers/models/time_series_transformers/configuration_time_series_transformer.py +34 -0
  80. optimum/rbln/transformers/models/time_series_transformers/modeling_time_series_transformers.py +52 -54
  81. optimum/rbln/transformers/models/wav2vec2/__init__.py +1 -0
  82. optimum/rbln/transformers/models/wav2vec2/configuration_wav2vec.py +19 -0
  83. optimum/rbln/transformers/models/wav2vec2/modeling_wav2vec2.py +9 -72
  84. optimum/rbln/transformers/models/whisper/__init__.py +1 -0
  85. optimum/rbln/transformers/models/whisper/configuration_whisper.py +64 -0
  86. optimum/rbln/transformers/models/whisper/modeling_whisper.py +57 -72
  87. optimum/rbln/transformers/models/xlm_roberta/__init__.py +1 -0
  88. optimum/rbln/transformers/models/xlm_roberta/configuration_xlm_roberta.py +19 -0
  89. optimum/rbln/transformers/models/xlm_roberta/modeling_xlm_roberta.py +3 -83
  90. optimum/rbln/utils/submodule.py +26 -43
  91. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a5.dist-info}/METADATA +1 -1
  92. optimum_rbln-0.7.4a5.dist-info/RECORD +162 -0
  93. optimum/rbln/modeling_config.py +0 -310
  94. optimum_rbln-0.7.4a4.dist-info/RECORD +0 -126
  95. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a5.dist-info}/WHEEL +0 -0
  96. {optimum_rbln-0.7.4a4.dist-info → optimum_rbln-0.7.4a5.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,196 @@ _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",
70
89
  "RBLNResNetForImageClassification",
71
- "RBLNXLMRobertaForSequenceClassification",
72
- "RBLNRobertaForSequenceClassification",
90
+ "RBLNResNetForImageClassificationConfig",
73
91
  "RBLNRobertaForMaskedLM",
74
- "RBLNViTForImageClassification",
75
- "RBLNBertForMaskedLM",
92
+ "RBLNRobertaForMaskedLMConfig",
93
+ "RBLNRobertaForSequenceClassification",
94
+ "RBLNRobertaForSequenceClassificationConfig",
95
+ "RBLNT5EncoderModel",
96
+ "RBLNT5EncoderModelConfig",
97
+ "RBLNT5ForConditionalGeneration",
98
+ "RBLNT5ForConditionalGenerationConfig",
76
99
  "RBLNTimeSeriesTransformerForPrediction",
100
+ "RBLNTimeSeriesTransformerForPredictionConfig",
101
+ "RBLNViTForImageClassification",
102
+ "RBLNViTForImageClassificationConfig",
103
+ "RBLNWav2Vec2ForCTC",
104
+ "RBLNWav2Vec2ForCTCConfig",
105
+ "RBLNWhisperForConditionalGeneration",
106
+ "RBLNWhisperForConditionalGenerationConfig",
107
+ "RBLNXLMRobertaForSequenceClassification",
108
+ "RBLNXLMRobertaForSequenceClassificationConfig",
109
+ "RBLNXLMRobertaModel",
110
+ "RBLNXLMRobertaModelConfig",
77
111
  ],
78
112
  "diffusers": [
79
113
  "RBLNAutoencoderKL",
114
+ "RBLNAutoencoderKLConfig",
80
115
  "RBLNControlNetModel",
81
- "RBLNPriorTransformer",
116
+ "RBLNControlNetModelConfig",
117
+ "RBLNDiffusionMixin",
82
118
  "RBLNKandinskyV22CombinedPipeline",
119
+ "RBLNKandinskyV22CombinedPipelineConfig",
83
120
  "RBLNKandinskyV22Img2ImgCombinedPipeline",
121
+ "RBLNKandinskyV22Img2ImgCombinedPipelineConfig",
122
+ "RBLNKandinskyV22Img2ImgPipeline",
123
+ "RBLNKandinskyV22Img2ImgPipelineConfig",
84
124
  "RBLNKandinskyV22InpaintCombinedPipeline",
125
+ "RBLNKandinskyV22InpaintCombinedPipelineConfig",
85
126
  "RBLNKandinskyV22InpaintPipeline",
86
- "RBLNKandinskyV22Img2ImgPipeline",
87
- "RBLNKandinskyV22PriorPipeline",
127
+ "RBLNKandinskyV22InpaintPipelineConfig",
88
128
  "RBLNKandinskyV22Pipeline",
89
- "RBLNStableDiffusionPipeline",
90
- "RBLNStableDiffusionXLPipeline",
91
- "RBLNUNet2DConditionModel",
92
- "RBLNStableDiffusionImg2ImgPipeline",
93
- "RBLNStableDiffusionInpaintPipeline",
94
- "RBLNStableDiffusionControlNetImg2ImgPipeline",
129
+ "RBLNKandinskyV22PipelineConfig",
130
+ "RBLNKandinskyV22PriorPipeline",
131
+ "RBLNKandinskyV22PriorPipelineConfig",
95
132
  "RBLNMultiControlNetModel",
96
- "RBLNStableDiffusionXLImg2ImgPipeline",
97
- "RBLNStableDiffusionXLInpaintPipeline",
98
- "RBLNStableDiffusionControlNetPipeline",
99
- "RBLNStableDiffusionXLControlNetPipeline",
100
- "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
101
- "RBLNVQModel",
133
+ "RBLNPriorTransformer",
134
+ "RBLNPriorTransformerConfig",
102
135
  "RBLNSD3Transformer2DModel",
136
+ "RBLNSD3Transformer2DModelConfig",
103
137
  "RBLNStableDiffusion3Img2ImgPipeline",
138
+ "RBLNStableDiffusion3Img2ImgPipelineConfig",
104
139
  "RBLNStableDiffusion3InpaintPipeline",
140
+ "RBLNStableDiffusion3InpaintPipelineConfig",
105
141
  "RBLNStableDiffusion3Pipeline",
106
- "RBLNDiffusionMixin",
142
+ "RBLNStableDiffusion3PipelineConfig",
143
+ "RBLNStableDiffusionControlNetImg2ImgPipeline",
144
+ "RBLNStableDiffusionControlNetImg2ImgPipelineConfig",
145
+ "RBLNStableDiffusionControlNetPipeline",
146
+ "RBLNStableDiffusionControlNetPipelineConfig",
147
+ "RBLNStableDiffusionImg2ImgPipeline",
148
+ "RBLNStableDiffusionImg2ImgPipelineConfig",
149
+ "RBLNStableDiffusionInpaintPipeline",
150
+ "RBLNStableDiffusionInpaintPipelineConfig",
151
+ "RBLNStableDiffusionPipeline",
152
+ "RBLNStableDiffusionPipelineConfig",
153
+ "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
154
+ "RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig",
155
+ "RBLNStableDiffusionXLControlNetPipeline",
156
+ "RBLNStableDiffusionXLControlNetPipelineConfig",
157
+ "RBLNStableDiffusionXLImg2ImgPipeline",
158
+ "RBLNStableDiffusionXLImg2ImgPipelineConfig",
159
+ "RBLNStableDiffusionXLInpaintPipeline",
160
+ "RBLNStableDiffusionXLInpaintPipelineConfig",
161
+ "RBLNStableDiffusionXLPipeline",
162
+ "RBLNStableDiffusionXLPipelineConfig",
163
+ "RBLNUNet2DConditionModel",
164
+ "RBLNUNet2DConditionModelConfig",
165
+ "RBLNVQModel",
166
+ "RBLNVQModelConfig",
107
167
  ],
108
168
  }
109
169
 
110
170
  if TYPE_CHECKING:
171
+ from .configuration_utils import (
172
+ RBLNCompileConfig,
173
+ RBLNModelConfig,
174
+ )
111
175
  from .diffusers import (
112
176
  RBLNAutoencoderKL,
177
+ RBLNAutoencoderKLConfig,
113
178
  RBLNControlNetModel,
179
+ RBLNControlNetModelConfig,
114
180
  RBLNDiffusionMixin,
115
181
  RBLNKandinskyV22CombinedPipeline,
182
+ RBLNKandinskyV22CombinedPipelineConfig,
116
183
  RBLNKandinskyV22Img2ImgCombinedPipeline,
184
+ RBLNKandinskyV22Img2ImgCombinedPipelineConfig,
117
185
  RBLNKandinskyV22Img2ImgPipeline,
186
+ RBLNKandinskyV22Img2ImgPipelineConfig,
118
187
  RBLNKandinskyV22InpaintCombinedPipeline,
188
+ RBLNKandinskyV22InpaintCombinedPipelineConfig,
119
189
  RBLNKandinskyV22InpaintPipeline,
190
+ RBLNKandinskyV22InpaintPipelineConfig,
120
191
  RBLNKandinskyV22Pipeline,
192
+ RBLNKandinskyV22PipelineConfig,
121
193
  RBLNKandinskyV22PriorPipeline,
194
+ RBLNKandinskyV22PriorPipelineConfig,
122
195
  RBLNMultiControlNetModel,
123
196
  RBLNPriorTransformer,
197
+ RBLNPriorTransformerConfig,
124
198
  RBLNSD3Transformer2DModel,
199
+ RBLNSD3Transformer2DModelConfig,
125
200
  RBLNStableDiffusion3Img2ImgPipeline,
201
+ RBLNStableDiffusion3Img2ImgPipelineConfig,
126
202
  RBLNStableDiffusion3InpaintPipeline,
203
+ RBLNStableDiffusion3InpaintPipelineConfig,
127
204
  RBLNStableDiffusion3Pipeline,
205
+ RBLNStableDiffusion3PipelineConfig,
128
206
  RBLNStableDiffusionControlNetImg2ImgPipeline,
207
+ RBLNStableDiffusionControlNetImg2ImgPipelineConfig,
129
208
  RBLNStableDiffusionControlNetPipeline,
209
+ RBLNStableDiffusionControlNetPipelineConfig,
130
210
  RBLNStableDiffusionImg2ImgPipeline,
211
+ RBLNStableDiffusionImg2ImgPipelineConfig,
131
212
  RBLNStableDiffusionInpaintPipeline,
213
+ RBLNStableDiffusionInpaintPipelineConfig,
132
214
  RBLNStableDiffusionPipeline,
215
+ RBLNStableDiffusionPipelineConfig,
133
216
  RBLNStableDiffusionXLControlNetImg2ImgPipeline,
217
+ RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig,
134
218
  RBLNStableDiffusionXLControlNetPipeline,
219
+ RBLNStableDiffusionXLControlNetPipelineConfig,
135
220
  RBLNStableDiffusionXLImg2ImgPipeline,
221
+ RBLNStableDiffusionXLImg2ImgPipelineConfig,
136
222
  RBLNStableDiffusionXLInpaintPipeline,
223
+ RBLNStableDiffusionXLInpaintPipelineConfig,
137
224
  RBLNStableDiffusionXLPipeline,
225
+ RBLNStableDiffusionXLPipelineConfig,
138
226
  RBLNUNet2DConditionModel,
227
+ RBLNUNet2DConditionModelConfig,
139
228
  RBLNVQModel,
229
+ RBLNVQModelConfig,
140
230
  )
141
231
  from .modeling import (
142
232
  RBLNBaseModel,
143
233
  RBLNModel,
144
234
  )
145
- from .modeling_config import (
146
- RBLNCompileConfig,
147
- RBLNConfig,
148
- )
149
235
  from .transformers import (
150
236
  RBLNASTForAudioClassification,
237
+ RBLNASTForAudioClassificationConfig,
151
238
  RBLNAutoModel,
152
239
  RBLNAutoModelForAudioClassification,
153
240
  RBLNAutoModelForCausalLM,
@@ -161,36 +248,69 @@ if TYPE_CHECKING:
161
248
  RBLNAutoModelForSpeechSeq2Seq,
162
249
  RBLNAutoModelForVision2Seq,
163
250
  RBLNBartForConditionalGeneration,
251
+ RBLNBartForConditionalGenerationConfig,
164
252
  RBLNBartModel,
253
+ RBLNBartModelConfig,
165
254
  RBLNBertForMaskedLM,
255
+ RBLNBertForMaskedLMConfig,
166
256
  RBLNBertForQuestionAnswering,
257
+ RBLNBertForQuestionAnsweringConfig,
167
258
  RBLNBertModel,
259
+ RBLNBertModelConfig,
168
260
  RBLNCLIPTextModel,
261
+ RBLNCLIPTextModelConfig,
169
262
  RBLNCLIPTextModelWithProjection,
263
+ RBLNCLIPTextModelWithProjectionConfig,
170
264
  RBLNCLIPVisionModel,
265
+ RBLNCLIPVisionModelConfig,
171
266
  RBLNCLIPVisionModelWithProjection,
267
+ RBLNCLIPVisionModelWithProjectionConfig,
268
+ RBLNDecoderOnlyModelForCausalLM,
269
+ RBLNDecoderOnlyModelForCausalLMConfig,
172
270
  RBLNDistilBertForQuestionAnswering,
271
+ RBLNDistilBertForQuestionAnsweringConfig,
173
272
  RBLNDPTForDepthEstimation,
273
+ RBLNDPTForDepthEstimationConfig,
174
274
  RBLNExaoneForCausalLM,
275
+ RBLNExaoneForCausalLMConfig,
175
276
  RBLNGemmaForCausalLM,
277
+ RBLNGemmaForCausalLMConfig,
176
278
  RBLNGPT2LMHeadModel,
279
+ RBLNGPT2LMHeadModelConfig,
177
280
  RBLNLlamaForCausalLM,
281
+ RBLNLlamaForCausalLMConfig,
178
282
  RBLNLlavaNextForConditionalGeneration,
283
+ RBLNLlavaNextForConditionalGenerationConfig,
179
284
  RBLNMidmLMHeadModel,
285
+ RBLNMidmLMHeadModelConfig,
180
286
  RBLNMistralForCausalLM,
287
+ RBLNMistralForCausalLMConfig,
181
288
  RBLNPhiForCausalLM,
289
+ RBLNPhiForCausalLMConfig,
182
290
  RBLNQwen2ForCausalLM,
291
+ RBLNQwen2ForCausalLMConfig,
183
292
  RBLNResNetForImageClassification,
293
+ RBLNResNetForImageClassificationConfig,
184
294
  RBLNRobertaForMaskedLM,
295
+ RBLNRobertaForMaskedLMConfig,
185
296
  RBLNRobertaForSequenceClassification,
297
+ RBLNRobertaForSequenceClassificationConfig,
186
298
  RBLNT5EncoderModel,
299
+ RBLNT5EncoderModelConfig,
187
300
  RBLNT5ForConditionalGeneration,
301
+ RBLNT5ForConditionalGenerationConfig,
188
302
  RBLNTimeSeriesTransformerForPrediction,
303
+ RBLNTimeSeriesTransformerForPredictionConfig,
189
304
  RBLNViTForImageClassification,
305
+ RBLNViTForImageClassificationConfig,
190
306
  RBLNWav2Vec2ForCTC,
307
+ RBLNWav2Vec2ForCTCConfig,
191
308
  RBLNWhisperForConditionalGeneration,
309
+ RBLNWhisperForConditionalGenerationConfig,
192
310
  RBLNXLMRobertaForSequenceClassification,
311
+ RBLNXLMRobertaForSequenceClassificationConfig,
193
312
  RBLNXLMRobertaModel,
313
+ RBLNXLMRobertaModelConfig,
194
314
  )
195
315
 
196
316
  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'
20
+ __version__ = version = '0.7.4a5'
21
21
  __version_tuple__ = version_tuple = (0, 7, 4)