From f9f1d71b384d3dec04bdfc1175fd4d769b72b6f3 Mon Sep 17 00:00:00 2001 From: Jing Jin <8752427+jinjingforever@users.noreply.github.com> Date: Mon, 28 Apr 2025 16:49:03 -0700 Subject: [PATCH] Update tasks and models. --- .../src/main/java/com/google/aiedge/gallery/data/Model.kt | 4 ++-- .../src/main/java/com/google/aiedge/gallery/data/Tasks.kt | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Model.kt b/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Model.kt index 7b99103..8a8d2a1 100644 --- a/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Model.kt +++ b/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Model.kt @@ -392,8 +392,8 @@ val MODELS_IMAGE_CLASSIFICATION: MutableList = mutableListOf( ) val MODELS_LLM: MutableList = mutableListOf( - MODEL_LLM_GEMMA_2B_GPU_INT4, - MODEL_LLM_GEMMA_2_2B_GPU_INT8, +// MODEL_LLM_GEMMA_2B_GPU_INT4, +// MODEL_LLM_GEMMA_2_2B_GPU_INT8, MODEL_LLM_GEMMA_3_1B_INT4, MODEL_LLM_DEEPSEEK, ) diff --git a/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Tasks.kt b/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Tasks.kt index 4d853dd..6caac37 100644 --- a/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Tasks.kt +++ b/Android/src/app/src/main/java/com/google/aiedge/gallery/data/Tasks.kt @@ -121,10 +121,10 @@ val TASK_IMAGE_GENERATION = Task( /** All tasks. */ val TASKS: List = listOf( // TASK_TEXT_CLASSIFICATION, - TASK_IMAGE_CLASSIFICATION, - TASK_IMAGE_GENERATION, - TASK_LLM_CHAT, +// TASK_IMAGE_CLASSIFICATION, +// TASK_IMAGE_GENERATION, TASK_LLM_SINGLE_TURN, + TASK_LLM_CHAT, ) fun getModelByName(name: String): Model? {