Update config.json

config.json

@@ -0,0 +1,119 @@
+{
+  "name": "Gemma🔥",
+  "schema_version": "latest",
+  "metadata": {
+    "parameters": {},
+    "models": {},
+    "default_model": null,
+    "model_parsers": null
+  },
+  "description": "**Test Gemma (7b and 2b) against Mistral-7B and TinyLLama**\nWant to try another model? Click '+' to add a cell and enter a Hugging Face model. \n",
+  "prompts": [
+    {
+      "name": "gemma-7b-prompt",
+      "input": "I went to the market and bought 10 apples. I gave 2 apples to the neighbor and 2 to the repairman. I then went and bought 5 more apples and ate 1. I also gave 3 bananas to my brother. How many apples did I remain with? Let's think step by step. ",
+      "metadata": {
+        "model": {
+          "name": "Text Generation",
+          "settings": {
+            "model": "google/gemma-7b",
+            "max_new_tokens": 500
+          }
+        },
+        "tags": null,
+        "parameters": {}
+      },
+      "outputs": [
+        {
+          "output_type": "execute_result",
+          "execution_count": 0,
+          "data": "10 apples were bought. 2 were given to the neighbor and 2 to the repairman. 10-2-2=6 apples were left. 5 more apples were bought. 6+5=11 apples were left. 1 was eaten. 11-1=10 apples were left. 3 bananas were given to the brother. 10-3=7 apples were left. The answer is 7 apples.\n\nAnswer:\n\nStep 1/2\n1. The person bought 10 apples.\n\nStep 2/2\n2. They gave 2 apples to the neighbor and 2 to the repairman. So, they have 10 - 2 - 2 = 6 apples left. Now, they bought 5 more apples. So, they have 6 + 5 = 11 apples. They ate 1 apple, so they have 11 - 1 = 10 apples. They gave 3 bananas to their brother. So, they have 10 - 3 = 7 apples left. Therefore, the person has $\\boxed{7}$ apples left.<eos>",
+          "mime_type": null,
+          "metadata": {}
+        }
+      ]
+    },
+    {
+      "name": "gemma-2b-prompt",
+      "input": "I went to the market and bought 10 apples. I gave 2 apples to the neighbor and 2 to the repairman. I then went and bought 5 more apples and ate 1. I also gave 3 bananas to my brother. How many apples did I remain with? Let's think step by step. ",
+      "metadata": {
+        "model": {
+          "name": "Text Generation",
+          "settings": {
+            "model": "google/gemma-2b",
+            "max_new_tokens": 500
+          }
+        },
+        "tags": null,
+        "parameters": {}
+      },
+      "outputs": [
+        {
+          "output_type": "execute_result",
+          "execution_count": 0,
+          "data": "10 - 2 - 2 = 6 6 + 5 = 11 11 - 1 = 10\n\nAnswer:\n\nStep 1/5\n1. I bought 10 apples and gave 2 to the neighbor and 2 to the repairman. So, I have 10 - 2 - 2 = 6 apples left.\n\nStep 2/5\n2. I then bought 5 more apples and ate 1. So, I have 6 + 5 = 11 apples left.\n\nStep 3/5\n3. I also gave 3 bananas to my brother. So, I have 11 - 3 = 8 apples left.\n\nStep 4/5\n4. I don't know how many apples I have left, but I know that I have 8 apples left.\n\nStep 5/5\n5. Therefore, the answer is 8 apples.<eos>",
+          "mime_type": null,
+          "metadata": {}
+        }
+      ]
+    },
+    {
+      "name": "mistral-prompt",
+      "input": "I went to the market and bought 10 apples. I gave 2 apples to the neighbor and 2 to the repairman. I then went and bought 5 more apples and ate 1. I also gave 3 bananas to my brother. How many apples did I remain with? Let's think step by step. ",
+      "metadata": {
+        "model": {
+          "name": "Text Generation",
+          "settings": {
+            "model": "mistralai/Mistral-7B-Instruct-v0.1"
+          }
+        },
+        "tags": null,
+        "parameters": {}
+      },
+      "outputs": [
+        {
+          "output_type": "execute_result",
+          "execution_count": 0,
+          "data": "\n\n1. I started with 10 apples. \n2. I gave away 2 apples to the neighbor and 2 to the repairman, so I have 10 - 4 = 6 apples left.\n3. Then I bought 5 more apples, making my total 6 + 5 = 11 apples.\n4. I ate 1 apple, leaving me with 11 - 1 = 10 apples.\n5. Finally, I gave 3 bananas to my brother, but this does not affect the number of apples I have.\n\nSo, I remained with 10 apples.</s>",
+          "mime_type": null,
+          "metadata": {}
+        }
+      ]
+    },
+    {
+      "name": "tinyllama-prompt",
+      "input": "I went to the market and bought 10 apples. I gave 2 apples to the neighbor and 2 to the repairman. I then went and bought 5 more apples and ate 1. I also gave 3 bananas to my brother. How many apples did I remain with? Let's think step by step. ",
+      "metadata": {
+        "model": {
+          "name": "Conversational",
+          "settings": {
+            "model": "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+          }
+        },
+        "tags": null,
+        "parameters": {}
+      },
+      "outputs": [
+        {
+          "output_type": "execute_result",
+          "execution_count": 0,
+          "data": "To calculate how many apples you have left, you need to subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought.\n\nIn this case, you bought 10 apples, so the total number of apples you bought is 10.\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given",
+          "mime_type": null,
+          "metadata": {
+            "raw_response": {
+              "generated_text": "To calculate how many apples you have left, you need to subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought.\n\nIn this case, you bought 10 apples, so the total number of apples you bought is 10.\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given",
+              "conversation": {
+                "generated_responses": [
+                  "To calculate how many apples you have left, you need to subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought.\n\nIn this case, you bought 10 apples, so the total number of apples you bought is 10.\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nNow, let's subtract the number of apples you gave to the neighbor and the repairman from the total number of apples you bought:\n\n- 2 apples given to the neighbor\n- 2 apples given to the repairman\n\nSo, the total number of apples you gave to the neighbor and the repairman is:\n\n- 2 apples given to the neighbor\n- 2 apples given"
+                ],
+                "past_user_inputs": [
+                  "I went to the market and bought 10 apples. I gave 2 apples to the neighbor and 2 to the repairman. I then went and bought 5 more apples and ate 1. I also gave 3 bananas to my brother. How many apples did I remain with? Let's think step by step. "
+                ]
+              }
+            }
+          }
+        }
+      ]
+    }
+  ]
+}