[mob] Fix tensor size for clip image

2024-08-20 12:54:26 +05:30
parent 692589e4c4
commit 3a0d6e808b
1 changed files with 2 additions and 1 deletions
--- a/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
+++ b/mobile/plugins/onnx_dart/android/src/main/kotlin/io/ente/photos/onnx_dart/OnnxDartPlugin.kt
@@ -178,7 +178,8 @@ class OnnxDartPlugin: FlutterPlugin, MethodCallHandler {
            inputTensorShape = longArrayOf(totalSize, 112, 112, 3)
          }
          ModelType.ClipImageEncoder -> {
-            inputTensorShape = longArrayOf(1, 3, 224, 224)
+            inputTensorShape = longArrayOf(1, 3, 256, 256)
+//            inputTensorShape = longArrayOf(1, 3, 256, 256)
          }
          ModelType.ClipTextEncoder -> {
            inputTensorShape = longArrayOf(1, 77)