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? {