archives/edge-gallery
archives/edge-gallery
2
0
Fork 0
mirror of https://github.com/google-ai-edge/gallery.git synced 2025-07-07 07:00:28 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update tasks and models.

Browse source
This commit is contained in:
Jing Jin 2025-04-28 16:49:03 -07:00
parent d0beaab31e
commit f9f1d71b38
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

4
Android/src/app/src/main/java/com/google/aiedge/gallery/data/Model.kt
View file

@ -392,8 +392,8 @@ val MODELS_IMAGE_CLASSIFICATION: MutableList<Model> = mutableListOf(
) )
val MODELS_LLM: MutableList<Model> = mutableListOf( val MODELS_LLM: MutableList<Model> = mutableListOf(
MODEL_LLM_GEMMA_2B_GPU_INT4, // MODEL_LLM_GEMMA_2B_GPU_INT4,
MODEL_LLM_GEMMA_2_2B_GPU_INT8, // MODEL_LLM_GEMMA_2_2B_GPU_INT8,
MODEL_LLM_GEMMA_3_1B_INT4, MODEL_LLM_GEMMA_3_1B_INT4,
MODEL_LLM_DEEPSEEK, MODEL_LLM_DEEPSEEK,
) )

6
Android/src/app/src/main/java/com/google/aiedge/gallery/data/Tasks.kt
View file

@ -121,10 +121,10 @@ val TASK_IMAGE_GENERATION = Task(
/** All tasks. */ /** All tasks. */
val TASKS: List<Task> = listOf( val TASKS: List<Task> = listOf(
// TASK_TEXT_CLASSIFICATION, // TASK_TEXT_CLASSIFICATION,
TASK_IMAGE_CLASSIFICATION, // TASK_IMAGE_CLASSIFICATION,
TASK_IMAGE_GENERATION, // TASK_IMAGE_GENERATION,
TASK_LLM_CHAT,
TASK_LLM_SINGLE_TURN, TASK_LLM_SINGLE_TURN,
TASK_LLM_CHAT,
) )
fun getModelByName(name: String): Model? { fun getModelByName(name: String): Model? {