fix: fix typos

unit-mesh · Nov 9, 2023 · 373e98f · 373e98f
1 parent 401f55c
commit 373e98f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/llm-modules/sentence-transformers/src/main/kotlin/cc/unitmesh/cf/STSemantic.kt b/llm-modules/sentence-transformers/src/main/kotlin/cc/unitmesh/cf/STSemantic.kt
@@ -46,11 +46,11 @@ class STSemantic(
  // floatArray is an inputIds.size * 384 array, we need to mean it to 384 * 1
  // 1, shape, shape.length
  val shapeSize = outputTensor.info.shape[2].toInt()
- val meanArray = FloatArray(shapeSzie)
- for (i in 0 until shapeSzie) {
+ val meanArray = FloatArray(shapeSize)
+ for (i in 0 until shapeSize) {
  var sum = 0f
  for (j in inputIds.indices) {
- sum += floatArray[j * shapeSzie + i]
+ sum += floatArray[j * shapeSize + i]
  }
 
  meanArray[i] = sum / inputIds.size