optimum-rbln 0.8.1a3__tar.gz → 0.8.1a4__tar.gz

{optimum_rbln-0.8.1a3 → optimum_rbln-0.8.1a4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: optimum-rbln
-Version: 0.8.1a3
+Version: 0.8.1a4
 Summary: Optimum RBLN is the interface between the HuggingFace Transformers and Diffusers libraries and RBLN accelerators. It provides a set of tools enabling easy model loading and inference on single and multiple rbln device settings for different downstream tasks.
 Project-URL: Homepage, https://rebellions.ai
 Project-URL: Documentation, https://docs.rbln.ai
@@ -23,7 +23,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: <3.13,>=3.9
 Requires-Dist: accelerate>=1.0.1
-Requires-Dist: diffusers<=0.31.0
+Requires-Dist: diffusers==0.34.0
 Requires-Dist: packaging>=24.1
 Requires-Dist: torch==2.6.0
 Requires-Dist: torchaudio<=2.6.0

{optimum_rbln-0.8.1a3 → optimum_rbln-0.8.1a4}/pyproject.toml

@@ -33,7 +33,7 @@ dependencies = [
     "torchvision<=0.21.0",
     "accelerate>=1.0.1",
     "transformers==4.51.3",
-    "diffusers<=0.31.0",
+    "diffusers==0.34.0",
     "packaging>=24.1",
 ]
 

{optimum_rbln-0.8.1a3 → optimum_rbln-0.8.1a4}/src/optimum/rbln/__version__.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.8.1a3'
-__version_tuple__ = version_tuple = (0, 8, 1, 'a3')
+__version__ = version = '0.8.1a4'
+__version_tuple__ = version_tuple = (0, 8, 1, 'a4')

{optimum_rbln-0.8.1a3 → optimum_rbln-0.8.1a4}/src/optimum/rbln/diffusers/models/unets/unet_2d_condition.py

@@ -185,7 +185,10 @@ class RBLNUNet2DConditionModel(RBLNModel):
         rbln_config: RBLNUNet2DConditionModelConfig,
         image_size: Optional[Tuple[int, int]] = None,
     ) -> Tuple[int, int]:
-        scale_factor = pipe.movq_scale_factor if hasattr(pipe, "movq_scale_factor") else pipe.vae_scale_factor
+        if hasattr(pipe, "movq"):
+            scale_factor = 2 ** (len(pipe.movq.config.block_out_channels) - 1)
+        else:
+            scale_factor = pipe.vae_scale_factor
 
         if image_size is None:
             if "Img2Img" in pipe.__class__.__name__:

{optimum_rbln-0.8.1a3 → optimum_rbln-0.8.1a4}/src/optimum/rbln/transformers/models/gemma3/modeling_gemma3.py

@@ -11,7 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import importlib
 import inspect
 from collections import deque
 from dataclasses import dataclass
@@ -124,23 +123,6 @@ class RBLNGemma3ForConditionalGeneration(RBLNModel):
     def can_generate(self):
         return True
 
-    @classmethod
-    def get_pytorch_model(cls, *args, **kwargs):
-        model = super().get_pytorch_model(*args, **kwargs)
-
-        with no_init_weights():
-            model_cls_name = model.model.language_model.__class__.__name__
-            causal_model_cls_name = model_cls_name.replace("TextModel", "ForCausalLM")
-            causal_model_cls = getattr(importlib.import_module("transformers"), causal_model_cls_name)
-            new_language_model = causal_model_cls(model.model.language_model.config)
-
-        new_language_model.lm_head = model.lm_head
-        new_language_model.model = model.model.language_model
-        model.model.language_model = new_language_model
-        model.lm_head = None
-        del model.lm_head
-        return model
-
     def __post_init__(self, **kwargs):
         self.vision_tower = LoopVisionTower(self.rbln_submodules[0])
         self.language_model = self.rbln_submodules[1]