optimum-rbln 0.7.3.post1__py3-none-any.whl → 0.7.4__py3-none-any.whl

optimum/rbln/__init__.py

@@ -25,11 +25,13 @@ _import_structure = {
         "RBLNBaseModel",
         "RBLNModel",
     ],
-    "modeling_config": [
+    "configuration_utils": [
         "RBLNCompileConfig",
-        "RBLNConfig",
+        "RBLNModelConfig",
     ],
     "transformers": [
+        "RBLNASTForAudioClassification",
+        "RBLNASTForAudioClassificationConfig",
         "RBLNAutoModel",
         "RBLNAutoModelForAudioClassification",
         "RBLNAutoModelForCausalLM",
@@ -43,110 +45,204 @@ _import_structure = {
         "RBLNAutoModelForSpeechSeq2Seq",
         "RBLNAutoModelForVision2Seq",
         "RBLNBartForConditionalGeneration",
+        "RBLNBartForConditionalGenerationConfig",
         "RBLNBartModel",
+        "RBLNBartModelConfig",
+        "RBLNBertForMaskedLM",
+        "RBLNBertForMaskedLMConfig",
+        "RBLNBertForQuestionAnswering",
+        "RBLNBertForQuestionAnsweringConfig",
         "RBLNBertModel",
+        "RBLNBertModelConfig",
         "RBLNCLIPTextModel",
+        "RBLNCLIPTextModelConfig",
         "RBLNCLIPTextModelWithProjection",
+        "RBLNCLIPTextModelWithProjectionConfig",
         "RBLNCLIPVisionModel",
+        "RBLNCLIPVisionModelConfig",
         "RBLNCLIPVisionModelWithProjection",
+        "RBLNCLIPVisionModelWithProjectionConfig",
+        "RBLNDecoderOnlyModelForCausalLM",
+        "RBLNDecoderOnlyModelForCausalLMConfig",
+        "RBLNDistilBertForQuestionAnswering",
+        "RBLNDistilBertForQuestionAnsweringConfig",
         "RBLNDPTForDepthEstimation",
+        "RBLNDPTForDepthEstimationConfig",
         "RBLNExaoneForCausalLM",
+        "RBLNExaoneForCausalLMConfig",
         "RBLNGemmaForCausalLM",
+        "RBLNGemmaForCausalLMConfig",
         "RBLNGPT2LMHeadModel",
-        "RBLNQwen2ForCausalLM",
-        "RBLNWav2Vec2ForCTC",
+        "RBLNGPT2LMHeadModelConfig",
+        "RBLNIdefics3VisionTransformer",
+        "RBLNIdefics3ForConditionalGeneration",
+        "RBLNIdefics3ForConditionalGenerationConfig",
+        "RBLNIdefics3VisionTransformerConfig",
         "RBLNLlamaForCausalLM",
-        "RBLNT5EncoderModel",
-        "RBLNT5ForConditionalGeneration",
-        "RBLNPhiForCausalLM",
+        "RBLNLlamaForCausalLMConfig",
         "RBLNLlavaNextForConditionalGeneration",
+        "RBLNLlavaNextForConditionalGenerationConfig",
         "RBLNMidmLMHeadModel",
+        "RBLNMidmLMHeadModelConfig",
         "RBLNMistralForCausalLM",
-        "RBLNWhisperForConditionalGeneration",
-        "RBLNXLMRobertaModel",
-        "RBLNASTForAudioClassification",
-        "RBLNBertForQuestionAnswering",
-        "RBLNDistilBertForQuestionAnswering",
+        "RBLNMistralForCausalLMConfig",
+        "RBLNPhiForCausalLM",
+        "RBLNPhiForCausalLMConfig",
+        "RBLNQwen2ForCausalLM",
+        "RBLNQwen2ForCausalLMConfig",
+        "RBLNQwen2_5_VisionTransformerPretrainedModel",
+        "RBLNQwen2_5_VisionTransformerPretrainedModelConfig",
+        "RBLNQwen2_5_VLForConditionalGeneration",
+        "RBLNQwen2_5_VLForConditionalGenerationConfig",
         "RBLNResNetForImageClassification",
-        "RBLNXLMRobertaForSequenceClassification",
-        "RBLNRobertaForSequenceClassification",
+        "RBLNResNetForImageClassificationConfig",
         "RBLNRobertaForMaskedLM",
+        "RBLNRobertaForMaskedLMConfig",
+        "RBLNRobertaForSequenceClassification",
+        "RBLNRobertaForSequenceClassificationConfig",
+        "RBLNT5EncoderModel",
+        "RBLNT5EncoderModelConfig",
+        "RBLNT5ForConditionalGeneration",
+        "RBLNT5ForConditionalGenerationConfig",
+        "RBLNTimeSeriesTransformerForPrediction",
+        "RBLNTimeSeriesTransformerForPredictionConfig",
         "RBLNViTForImageClassification",
-        "RBLNBertForMaskedLM",
+        "RBLNViTForImageClassificationConfig",
+        "RBLNWav2Vec2ForCTC",
+        "RBLNWav2Vec2ForCTCConfig",
+        "RBLNWhisperForConditionalGeneration",
+        "RBLNWhisperForConditionalGenerationConfig",
+        "RBLNXLMRobertaForSequenceClassification",
+        "RBLNXLMRobertaForSequenceClassificationConfig",
+        "RBLNXLMRobertaModel",
+        "RBLNXLMRobertaModelConfig",
     ],
     "diffusers": [
         "RBLNAutoencoderKL",
+        "RBLNAutoencoderKLConfig",
         "RBLNControlNetModel",
-        "RBLNPriorTransformer",
+        "RBLNControlNetModelConfig",
+        "RBLNDiffusionMixin",
         "RBLNKandinskyV22CombinedPipeline",
+        "RBLNKandinskyV22CombinedPipelineConfig",
         "RBLNKandinskyV22Img2ImgCombinedPipeline",
+        "RBLNKandinskyV22Img2ImgCombinedPipelineConfig",
+        "RBLNKandinskyV22Img2ImgPipeline",
+        "RBLNKandinskyV22Img2ImgPipelineConfig",
         "RBLNKandinskyV22InpaintCombinedPipeline",
+        "RBLNKandinskyV22InpaintCombinedPipelineConfig",
         "RBLNKandinskyV22InpaintPipeline",
-        "RBLNKandinskyV22Img2ImgPipeline",
-        "RBLNKandinskyV22PriorPipeline",
+        "RBLNKandinskyV22InpaintPipelineConfig",
         "RBLNKandinskyV22Pipeline",
-        "RBLNStableDiffusionPipeline",
-        "RBLNStableDiffusionXLPipeline",
-        "RBLNUNet2DConditionModel",
-        "RBLNStableDiffusionImg2ImgPipeline",
-        "RBLNStableDiffusionInpaintPipeline",
-        "RBLNStableDiffusionControlNetImg2ImgPipeline",
+        "RBLNKandinskyV22PipelineConfig",
+        "RBLNKandinskyV22PriorPipeline",
+        "RBLNKandinskyV22PriorPipelineConfig",
         "RBLNMultiControlNetModel",
-        "RBLNStableDiffusionXLImg2ImgPipeline",
-        "RBLNStableDiffusionXLInpaintPipeline",
-        "RBLNStableDiffusionControlNetPipeline",
-        "RBLNStableDiffusionXLControlNetPipeline",
-        "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
-        "RBLNVQModel",
+        "RBLNPriorTransformer",
+        "RBLNPriorTransformerConfig",
         "RBLNSD3Transformer2DModel",
+        "RBLNSD3Transformer2DModelConfig",
         "RBLNStableDiffusion3Img2ImgPipeline",
+        "RBLNStableDiffusion3Img2ImgPipelineConfig",
         "RBLNStableDiffusion3InpaintPipeline",
+        "RBLNStableDiffusion3InpaintPipelineConfig",
         "RBLNStableDiffusion3Pipeline",
-        "RBLNDiffusionMixin",
+        "RBLNStableDiffusion3PipelineConfig",
+        "RBLNStableDiffusionControlNetImg2ImgPipeline",
+        "RBLNStableDiffusionControlNetImg2ImgPipelineConfig",
+        "RBLNStableDiffusionControlNetPipeline",
+        "RBLNStableDiffusionControlNetPipelineConfig",
+        "RBLNStableDiffusionImg2ImgPipeline",
+        "RBLNStableDiffusionImg2ImgPipelineConfig",
+        "RBLNStableDiffusionInpaintPipeline",
+        "RBLNStableDiffusionInpaintPipelineConfig",
+        "RBLNStableDiffusionPipeline",
+        "RBLNStableDiffusionPipelineConfig",
+        "RBLNStableDiffusionXLControlNetImg2ImgPipeline",
+        "RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig",
+        "RBLNStableDiffusionXLControlNetPipeline",
+        "RBLNStableDiffusionXLControlNetPipelineConfig",
+        "RBLNStableDiffusionXLImg2ImgPipeline",
+        "RBLNStableDiffusionXLImg2ImgPipelineConfig",
+        "RBLNStableDiffusionXLInpaintPipeline",
+        "RBLNStableDiffusionXLInpaintPipelineConfig",
+        "RBLNStableDiffusionXLPipeline",
+        "RBLNStableDiffusionXLPipelineConfig",
+        "RBLNUNet2DConditionModel",
+        "RBLNUNet2DConditionModelConfig",
+        "RBLNVQModel",
+        "RBLNVQModelConfig",
     ],
 }
 
 if TYPE_CHECKING:
+    from .configuration_utils import (
+        RBLNCompileConfig,
+        RBLNModelConfig,
+    )
     from .diffusers import (
         RBLNAutoencoderKL,
+        RBLNAutoencoderKLConfig,
         RBLNControlNetModel,
+        RBLNControlNetModelConfig,
         RBLNDiffusionMixin,
         RBLNKandinskyV22CombinedPipeline,
+        RBLNKandinskyV22CombinedPipelineConfig,
         RBLNKandinskyV22Img2ImgCombinedPipeline,
+        RBLNKandinskyV22Img2ImgCombinedPipelineConfig,
         RBLNKandinskyV22Img2ImgPipeline,
+        RBLNKandinskyV22Img2ImgPipelineConfig,
         RBLNKandinskyV22InpaintCombinedPipeline,
+        RBLNKandinskyV22InpaintCombinedPipelineConfig,
         RBLNKandinskyV22InpaintPipeline,
+        RBLNKandinskyV22InpaintPipelineConfig,
         RBLNKandinskyV22Pipeline,
+        RBLNKandinskyV22PipelineConfig,
         RBLNKandinskyV22PriorPipeline,
+        RBLNKandinskyV22PriorPipelineConfig,
         RBLNMultiControlNetModel,
         RBLNPriorTransformer,
+        RBLNPriorTransformerConfig,
         RBLNSD3Transformer2DModel,
+        RBLNSD3Transformer2DModelConfig,
         RBLNStableDiffusion3Img2ImgPipeline,
+        RBLNStableDiffusion3Img2ImgPipelineConfig,
         RBLNStableDiffusion3InpaintPipeline,
+        RBLNStableDiffusion3InpaintPipelineConfig,
         RBLNStableDiffusion3Pipeline,
+        RBLNStableDiffusion3PipelineConfig,
         RBLNStableDiffusionControlNetImg2ImgPipeline,
+        RBLNStableDiffusionControlNetImg2ImgPipelineConfig,
         RBLNStableDiffusionControlNetPipeline,
+        RBLNStableDiffusionControlNetPipelineConfig,
         RBLNStableDiffusionImg2ImgPipeline,
+        RBLNStableDiffusionImg2ImgPipelineConfig,
         RBLNStableDiffusionInpaintPipeline,
+        RBLNStableDiffusionInpaintPipelineConfig,
         RBLNStableDiffusionPipeline,
+        RBLNStableDiffusionPipelineConfig,
         RBLNStableDiffusionXLControlNetImg2ImgPipeline,
+        RBLNStableDiffusionXLControlNetImg2ImgPipelineConfig,
         RBLNStableDiffusionXLControlNetPipeline,
+        RBLNStableDiffusionXLControlNetPipelineConfig,
         RBLNStableDiffusionXLImg2ImgPipeline,
+        RBLNStableDiffusionXLImg2ImgPipelineConfig,
         RBLNStableDiffusionXLInpaintPipeline,
+        RBLNStableDiffusionXLInpaintPipelineConfig,
         RBLNStableDiffusionXLPipeline,
+        RBLNStableDiffusionXLPipelineConfig,
         RBLNUNet2DConditionModel,
+        RBLNUNet2DConditionModelConfig,
         RBLNVQModel,
+        RBLNVQModelConfig,
     )
     from .modeling import (
         RBLNBaseModel,
         RBLNModel,
     )
-    from .modeling_config import (
-        RBLNCompileConfig,
-        RBLNConfig,
-    )
     from .transformers import (
         RBLNASTForAudioClassification,
+        RBLNASTForAudioClassificationConfig,
         RBLNAutoModel,
         RBLNAutoModelForAudioClassification,
         RBLNAutoModelForCausalLM,
@@ -160,35 +256,77 @@ if TYPE_CHECKING:
         RBLNAutoModelForSpeechSeq2Seq,
         RBLNAutoModelForVision2Seq,
         RBLNBartForConditionalGeneration,
+        RBLNBartForConditionalGenerationConfig,
         RBLNBartModel,
+        RBLNBartModelConfig,
         RBLNBertForMaskedLM,
+        RBLNBertForMaskedLMConfig,
         RBLNBertForQuestionAnswering,
+        RBLNBertForQuestionAnsweringConfig,
         RBLNBertModel,
+        RBLNBertModelConfig,
         RBLNCLIPTextModel,
+        RBLNCLIPTextModelConfig,
         RBLNCLIPTextModelWithProjection,
+        RBLNCLIPTextModelWithProjectionConfig,
         RBLNCLIPVisionModel,
+        RBLNCLIPVisionModelConfig,
         RBLNCLIPVisionModelWithProjection,
+        RBLNCLIPVisionModelWithProjectionConfig,
+        RBLNDecoderOnlyModelForCausalLM,
+        RBLNDecoderOnlyModelForCausalLMConfig,
         RBLNDistilBertForQuestionAnswering,
+        RBLNDistilBertForQuestionAnsweringConfig,
         RBLNDPTForDepthEstimation,
+        RBLNDPTForDepthEstimationConfig,
         RBLNExaoneForCausalLM,
+        RBLNExaoneForCausalLMConfig,
         RBLNGemmaForCausalLM,
+        RBLNGemmaForCausalLMConfig,
         RBLNGPT2LMHeadModel,
+        RBLNGPT2LMHeadModelConfig,
+        RBLNIdefics3ForConditionalGeneration,
+        RBLNIdefics3ForConditionalGenerationConfig,
+        RBLNIdefics3VisionTransformer,
+        RBLNIdefics3VisionTransformerConfig,
         RBLNLlamaForCausalLM,
+        RBLNLlamaForCausalLMConfig,
         RBLNLlavaNextForConditionalGeneration,
+        RBLNLlavaNextForConditionalGenerationConfig,
         RBLNMidmLMHeadModel,
+        RBLNMidmLMHeadModelConfig,
         RBLNMistralForCausalLM,
+        RBLNMistralForCausalLMConfig,
         RBLNPhiForCausalLM,
+        RBLNPhiForCausalLMConfig,
+        RBLNQwen2_5_VisionTransformerPretrainedModel,
+        RBLNQwen2_5_VisionTransformerPretrainedModelConfig,
+        RBLNQwen2_5_VLForConditionalGeneration,
+        RBLNQwen2_5_VLForConditionalGenerationConfig,
         RBLNQwen2ForCausalLM,
+        RBLNQwen2ForCausalLMConfig,
         RBLNResNetForImageClassification,
+        RBLNResNetForImageClassificationConfig,
         RBLNRobertaForMaskedLM,
+        RBLNRobertaForMaskedLMConfig,
         RBLNRobertaForSequenceClassification,
+        RBLNRobertaForSequenceClassificationConfig,
         RBLNT5EncoderModel,
+        RBLNT5EncoderModelConfig,
         RBLNT5ForConditionalGeneration,
+        RBLNT5ForConditionalGenerationConfig,
+        RBLNTimeSeriesTransformerForPrediction,
+        RBLNTimeSeriesTransformerForPredictionConfig,
         RBLNViTForImageClassification,
+        RBLNViTForImageClassificationConfig,
         RBLNWav2Vec2ForCTC,
+        RBLNWav2Vec2ForCTCConfig,
         RBLNWhisperForConditionalGeneration,
+        RBLNWhisperForConditionalGenerationConfig,
         RBLNXLMRobertaForSequenceClassification,
+        RBLNXLMRobertaForSequenceClassificationConfig,
         RBLNXLMRobertaModel,
+        RBLNXLMRobertaModelConfig,
     )
 
 else:

optimum/rbln/__version__.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.7.3.post1'
-__version_tuple__ = version_tuple = (0, 7, 3)
+__version__ = version = '0.7.4'
+__version_tuple__ = version_tuple = (0, 7, 4)