Update optimum/exporters/openvino/model_patcher.py

huggingface · Jul 10, 2024 · d82faa3 · d82faa3
1 parent 26d9593
commit d82faa3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/exporters/openvino/model_patcher.py b/optimum/exporters/openvino/model_patcher.py
@@ -539,7 +539,7 @@ def create_sinusoidal_positions(num_pos: int, dim: int, base: int = 10000) -> to
 
             base = self._model.model.layers[0].self_attn.rotary_emb.base
             dim = self._model.model.layers[0].self_attn.rotary_emb.dim
-            embed_positions = create_sinusoidal_positions(max_positions, dim, base),
+            embed_positions = create_sinusoidal_positions(max_positions, dim, base)
 
             for layer in self._model.model.layers:
                 layer.self_attn.rotary_emb.register_buffer("embed_positions", embed_positions)