Skip to content

Commit 290e947

Browse files
committed
Fix
1 parent 740bbac commit 290e947

File tree

3 files changed

+2
-4
lines changed

3 files changed

+2
-4
lines changed

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/BenchmarkActivity.java

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
import android.os.Looper;
1717
import android.system.ErrnoException;
1818
import android.system.Os;
19-
import android.util.Log;
2019
import com.google.gson.Gson;
2120
import java.io.File;
2221
import java.io.FileWriter;
@@ -127,7 +126,7 @@ public void handleMessage(android.os.Message msg) {
127126
mLlmBenchmark =
128127
new LlmBenchmark(
129128
mBenchmarkActivity,
130-
mBenchmarkActivity.mModel.getName(),
129+
mBenchmarkActivity.mModel.getPath(),
131130
mBenchmarkActivity.mTokenizerPath,
132131
mBenchmarkActivity.mPrompt,
133132
mBenchmarkActivity.mTemperature,

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/LlmBenchmark.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ public class LlmBenchmark implements LlmModelRunnerCallback {
3232
mResults = results;
3333
mActivity = activity;
3434
mStatsInfo = new StatsInfo();
35-
mStatsInfo.modelName = modelFile.replace(".pte", "");
35+
mStatsInfo.modelName = modelFile.substring(modelFile.lastIndexOf('/') + 1).replace(".pte", "");
3636
mPrompt = prompt;
3737
mLlmModelRunner = new LlmModelRunner(modelFile, tokenizerPath, temperature, this);
3838
mStatsInfo.loadStart = System.nanoTime();

extension/benchmark/android/benchmark/app/src/main/java/org/pytorch/minibench/ModelRunner.java

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
package org.pytorch.minibench;
1010

1111
import android.os.Debug;
12-
import android.util.Log;
1312
import java.io.File;
1413
import java.util.ArrayList;
1514
import java.util.Collections;

0 commit comments

Comments
 (0)