mirror of
https://github.com/google-ai-edge/gallery.git
synced 2025-07-06 06:30:30 -04:00
Update tasks and models.
This commit is contained in:
parent
d0beaab31e
commit
f9f1d71b38
2 changed files with 5 additions and 5 deletions
|
@ -392,8 +392,8 @@ val MODELS_IMAGE_CLASSIFICATION: MutableList<Model> = mutableListOf(
|
|||
)
|
||||
|
||||
val MODELS_LLM: MutableList<Model> = 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,
|
||||
)
|
||||
|
|
|
@ -121,10 +121,10 @@ val TASK_IMAGE_GENERATION = Task(
|
|||
/** All tasks. */
|
||||
val TASKS: List<Task> = 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? {
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue