Update tasks and models.

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

View file

@ -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,
)

View file

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