diff --git a/matrixbot/commands.py b/matrixbot/commands.py index b71ca65..ce45953 100644 --- a/matrixbot/commands.py +++ b/matrixbot/commands.py @@ -24,22 +24,24 @@ logger = logging.getLogger("matrixbot") # Human-readable display names for Ollama model tags _MODEL_DISPLAY = { - "lotusllm": "Llama 3.2 1B", - "lotusllm:latest": "Llama 3.2 1B", - "lotusllmben:latest": "Llama 2 7B", - "sadiq-bd/llama3.2-1b-uncensored:latest": "Llama 3.2 1B", - "huihui_ai/llama3.2-abliterate:3b": "Llama 3.2 3B", - "huihui_ai/llama3.2-abliterated:3b": "Llama 3.2 3B", - "huihui_ai/gemma3-abliterated:1b": "Gemma 3 1B", + "lotusllm": "Llama 3.2 1B (uncensored)", + "lotusllm:latest": "Llama 3.2 1B (uncensored)", + "lotusllmben:latest": "Llama 2 7B (uncensored)", + "sadiq-bd/llama3.2-1b-uncensored:latest": "Llama 3.2 1B (uncensored)", + "huihui_ai/llama3.2-abliterate:3b": "Llama 3.2 3B (abliterated)", + "huihui_ai/llama3.2-abliterated:3b": "Llama 3.2 3B (abliterated)", + "huihui_ai/gemma3-abliterated:1b": "Gemma 3 1B (abliterated)", + "llama2-uncensored:latest": "Llama 2 7B (uncensored)", + "llama2-uncensored-kevin:latest": "Llama 2 7B (uncensored)", "llama3.2:latest": "Llama 3.2 3B", "llama3.2:1b": "Llama 3.2 1B", "llama3.3:latest": "Llama 3.3 70B", "gemma3:latest": "Gemma 3 4B", "gemma3:1b": "Gemma 3 1B", "phi4-mini:latest": "Phi-4 Mini 3.8B", - "deepseek-r1:latest": "DeepSeek R1", + "deepseek-r1:latest": "DeepSeek R1 7B", "codellama:latest": "Code Llama 7B", - "dolphin-phi:latest": "Dolphin Phi", + "dolphin-phi:latest": "Dolphin Phi 2.7B (uncensored)", "qwen2.5:latest": "Qwen 2.5 7B", "qwen2.5:7b": "Qwen 2.5 7B", }