diff --git a/modules/dnn/src/layers/rotary_embedding_layer.cpp b/modules/dnn/src/layers/rotary_embedding_layer.cpp index 83d0e05a01..aa0beeb5a3 100644 --- a/modules/dnn/src/layers/rotary_embedding_layer.cpp +++ b/modules/dnn/src/layers/rotary_embedding_layer.cpp @@ -231,10 +231,10 @@ class RotaryEmbeddingLayerImpl CV_FINAL : public RotaryEmbeddingLayer { ); CV_CheckTrue( cos_cache_shape.dims == 3 || inputs.size() > 3, - "RotaryEmbeddingLayer: provide position_ids or specify sin_cache and cos_cahe in format BxTxD" + "RotaryEmbeddingLayer: provide position_ids or specify sin_cache and cos_cache in format BxTxD" ); CV_CheckTrue( - cos_cache_shape.dims == cos_cache_shape.dims, + cos_cache_shape.dims == sin_cache_shape.dims, "RotaryEmbeddingLayer: cos_cache and sin_cache must have the same number of dimensions" ); for (int i = 0; i < cos_cache_shape.dims; ++i) diff --git a/modules/dnn/src/onnx/onnx_importer2.cpp b/modules/dnn/src/onnx/onnx_importer2.cpp index 0762b3188e..da2dad5808 100644 --- a/modules/dnn/src/onnx/onnx_importer2.cpp +++ b/modules/dnn/src/onnx/onnx_importer2.cpp @@ -2726,6 +2726,8 @@ void ONNXImporter2::buildDispatchMap_ONNX_AI() dispatch["Tile"] = &ONNXImporter2::parseTile; dispatch["LayerNormalization"] = &ONNXImporter2::parseLayerNorm; dispatch["GroupNormalization"] = &ONNXImporter2::parseInstanceNormalization; + dispatch["RMSNormalization"] = &ONNXImporter2::parseRMSNormalization; + dispatch["RotaryEmbedding"] = &ONNXImporter2::parseRotaryEmbedding; dispatch["NegativeLogLikelihoodLoss"] = &ONNXImporter2::parseNegativeLogLikelihoodLoss; dispatch["SoftmaxCrossEntropyLoss"] = &ONNXImporter2::parseSoftmaxCrossEntropyLoss; // @TODO@ONNX: Add support for SDPA