Upload Mimma_3_1b_chat.ipynb

Mimma_3_1b_chat.ipynb

@@ -0,0 +1,197 @@
+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": [],
+      "gpuType": "T4"
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    },
+    "accelerator": "GPU"
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": 1,
+      "metadata": {
+        "id": "TFu_ibC1eYrz"
+      },
+      "outputs": [],
+      "source": [
+        "!pip install torch transformers -q"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "import torch\n",
+        "from transformers import pipeline\n",
+        "from IPython.display import clear_output\n",
+        "from google.colab import output"
+      ],
+      "metadata": {
+        "id": "Zs7QNs0Tet6r"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "class ChatBot:\n",
+        "    _instance = None\n",
+        "    _current_model = None\n",
+        "\n",
+        "    def __init__(self, model_slug=None):\n",
+        "        if model_slug and model_slug != ChatBot._current_model:\n",
+        "            self.load_model(model_slug)\n",
+        "            ChatBot._current_model = model_slug\n",
+        "\n",
+        "        self.messages = []\n",
+        "        self.max_tokens = 512\n",
+        "        self.temperature = 0.01\n",
+        "        self.top_k = 64\n",
+        "        self.top_p = 0.95\n",
+        "        self.min_p = 0.0\n",
+        "        self.repetition_penalty = 1.3\n",
+        "\n",
+        "    @classmethod\n",
+        "    def get_instance(cls, model_slug=None):\n",
+        "        if not cls._instance or (model_slug and model_slug != cls._current_model):\n",
+        "            cls._instance = cls(model_slug)\n",
+        "        return cls._instance\n",
+        "\n",
+        "    def load_model(self, model_slug):\n",
+        "        print(f\"Loading model {model_slug}...\")\n",
+        "        self.pipeline = pipeline(\n",
+        "            \"text-generation\",\n",
+        "            model=model_slug,\n",
+        "            device_map=\"cuda\",\n",
+        "            torch_dtype=torch.bfloat16\n",
+        "        )\n",
+        "        clear_output()\n",
+        "        print(\"Model loaded successfully!\")\n",
+        "\n",
+        "    def reset_conversation(self, system_message):\n",
+        "        \"\"\"Reset the conversation with a new system message\"\"\"\n",
+        "        self.messages = [{\"role\": \"system\", \"content\": system_message}]\n",
+        "\n",
+        "    def get_response(self, user_input):\n",
+        "        \"\"\"Get response with current parameters\"\"\"\n",
+        "        self.messages.append({\"role\": \"user\", \"content\": user_input})\n",
+        "        outputs = self.pipeline(\n",
+        "            self.messages,\n",
+        "            max_new_tokens=self.max_tokens,\n",
+        "            do_sample=True,\n",
+        "            temperature=self.temperature,\n",
+        "            top_k=self.top_k,\n",
+        "            top_p=self.top_p,\n",
+        "            min_p = self.min_p,\n",
+        "            repetition_penalty = self.repetition_penalty\n",
+        "        )\n",
+        "        response = outputs[0][\"generated_text\"][-1]\n",
+        "        content = response.get('content', 'No content available')\n",
+        "        self.messages.append({\"role\": \"assistant\", \"content\": content})\n",
+        "        return content\n",
+        "\n",
+        "    def update_params(self, max_tokens=None, temperature=None, top_k=None, top_p=None, min_p=None, repetition_penalty=None):\n",
+        "        \"\"\"Update generation parameters\"\"\"\n",
+        "        if max_tokens is not None:\n",
+        "            self.max_tokens = max_tokens\n",
+        "        if temperature is not None:\n",
+        "            self.temperature = temperature\n",
+        "        if top_k is not None:\n",
+        "            self.top_k = top_k\n",
+        "        if top_p is not None:\n",
+        "            self.top_p = top_p\n",
+        "        if min_p is not None:\n",
+        "            self.min_p = min_p\n",
+        "        if repetition_penalty is not None:\n",
+        "            self.repetition_penalty = repetition_penalty"
+      ],
+      "metadata": {
+        "id": "v4uIN6uIeyl3"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "def run_chatbot(\n",
+        "    model=None,\n",
+        "    system_message=\"You are Mimma, You are expert in python coding, Think step by step before coming up with final python code\",\n",
+        "    max_tokens=None,\n",
+        "    temperature=None,\n",
+        "    top_k=None,\n",
+        "    top_p=None,\n",
+        "):\n",
+        "    try:\n",
+        "        # Get or create chatbot instance\n",
+        "        chatbot = ChatBot.get_instance(model)\n",
+        "\n",
+        "        # Update parameters if provided\n",
+        "        if max_tokens or temperature or top_k or top_p:\n",
+        "          chatbot.update_params(max_tokens, temperature, top_k, top_p)\n",
+        "\n",
+        "        # Reset conversation with new system message\n",
+        "        if system_message:\n",
+        "          chatbot.reset_conversation(system_message)\n",
+        "\n",
+        "        print(\"Chatbot: Hi! Type 'quit' to exit.\")\n",
+        "\n",
+        "        while True:\n",
+        "            user_input = input(\"You: \").strip()\n",
+        "            if user_input.lower() == 'quit':\n",
+        "                break\n",
+        "            try:\n",
+        "                response = chatbot.get_response(user_input)\n",
+        "                print(\"Chatbot:\", response)\n",
+        "            except Exception as e:\n",
+        "                print(f\"Chatbot: An error occurred: {str(e)}\")\n",
+        "                print(\"Please try again.\")\n",
+        "\n",
+        "    except Exception as e:\n",
+        "        print(f\"Error in chatbot: {str(e)}\")"
+      ],
+      "metadata": {
+        "id": "H2n_6Xcue3Vn"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "run_chatbot(model=\"pankajmathur/Mimma-3-1b\")"
+      ],
+      "metadata": {
+        "id": "JEqgoAH2fC6h"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# # change system message\n",
+        "# run_chatbot(\n",
+        "#     system_message=\"You are Orca Mini, You are expert in logic, Think step by step before coming up with final answer\",\n",
+        "#     max_tokens=1024,\n",
+        "#     temperature=0.3\n",
+        "# )"
+      ],
+      "metadata": {
+        "id": "tGW8wsfAfHDf"
+      },
+      "execution_count": null,
+      "outputs": []
+    }
+  ]
+}