google-deepmind/gemma-2b-it:dff94eaf | Run with an API on Replicate

Version

You're looking at a specific version of this model. Jump to the model overview.

google-deepmind /gemma-2b-it:dff94eaf

Playground API

Input

prompt

string

Shift + Return to add a new line

Prompt to send to the model.

Default: "Write me a poem about Machine Learning."

max_new_tokens

integer

(minimum: 1)

Maximum number of tokens to generate. A word is generally 2-3 tokens

Default: 200

min_new_tokens

integer

(minimum: -1)

Minimum number of tokens to generate. To disable, set to -1. A word is generally 2-3 tokens.

Default: -1

temperature

number

(minimum: 0.01, maximum: 5)

Adjusts randomness of outputs, greater than 1 is random and 0 is deterministic, 0.75 is a good starting value.

Default: 0.7

top_p

number

(minimum: 0, maximum: 1)

When decoding text, samples from the top p percentage of most likely tokens; lower to ignore less likely tokens

Default: 0.95

top_k

integer

When decoding text, samples from the top k most likely tokens; lower to ignore less likely tokens

Default: 50

repetition_penalty

number

(minimum: 0)

A parameter that controls how repetitive text can be. Lower means more repetitive, while higher means less repetitive. Set to 1.0 to disable.

Default: 1.15

Run this model in Node.js with one line of code:

npx create-replicate --model=google-deepmind/gemma-2b-it

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run google-deepmind/gemma-2b-it using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "google-deepmind/gemma-2b-it:dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626",
  {
    input: {
      top_k: 50,
      top_p: 0.95,
      prompt: "Write me a poem about Machine Learning.",
      temperature: 0.7,
      max_new_tokens: 200,
      min_new_tokens: -1,
      repetition_penalty: 1.15
    }
  }
);

console.log(output);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run google-deepmind/gemma-2b-it using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "google-deepmind/gemma-2b-it:dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626",
    input={
        "top_k": 50,
        "top_p": 0.95,
        "prompt": "Write me a poem about Machine Learning.",
        "temperature": 0.7,
        "max_new_tokens": 200,
        "min_new_tokens": -1,
        "repetition_penalty": 1.15
    }
)

# The google-deepmind/gemma-2b-it model can stream output as it's running.
# The predict method returns an iterator, and you can iterate over that output.
for item in output:
    # https://replicate.com/google-deepmind/gemma-2b-it/api#output-schema
    print(item, end="")

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run google-deepmind/gemma-2b-it using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "google-deepmind/gemma-2b-it:dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626",
    "input": {
      "top_k": 50,
      "top_p": 0.95,
      "prompt": "Write me a poem about Machine Learning.",
      "temperature": 0.7,
      "max_new_tokens": 200,
      "min_new_tokens": -1,
      "repetition_penalty": 1.15
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

Machines learn, they never sleep, Algorithms spin, a pattern to keep. From data's vast and ever-growing sea, They weave insights, a knowledge to believe. Supervised learning, a guiding hand, Training data, shaping the sand. Unsupervised, finding hidden gems, Clustering data, revealing their fears. Reinforcement, a learning spree, Learning from mistakes, a path to believe. Deep learning, a hidden black box, Extracting patterns, where once there was no shock. The future holds, a world transformed, With machines learning, we are transformed. A symphony of algorithms, a human guide, Igniting a future, where possibilities reside.

{
  "completed_at": "2024-02-21T20:45:57.903047Z",
  "created_at": "2024-02-21T20:43:45.881563Z",
  "data_removed": false,
  "error": null,
  "id": "qgxsdm3blag42bc7qhceivtbma",
  "input": {
    "top_k": 50,
    "top_p": 0.95,
    "prompt": "Write me a poem about Machine Learning.",
    "temperature": 0.7,
    "max_new_tokens": 200,
    "min_new_tokens": -1,
    "repetition_penalty": 1.15
  },
  "logs": null,
  "metrics": {
    "predict_time": 5.049841,
    "total_time": 132.021484
  },
  "output": [
    "\n\n",
    "",
    "Machines ",
    "",
    "learn, ",
    "they ",
    "never ",
    "",
    "sleep,\n",
    "",
    "Algorithms ",
    "",
    "spin, ",
    "a ",
    "pattern ",
    "to ",
    "",
    "keep.\n",
    "",
    "From ",
    "",
    "",
    "data's ",
    "vast ",
    "and ",
    "",
    "",
    "ever-growing ",
    "",
    "sea,\n",
    "",
    "They ",
    "weave ",
    "",
    "insights, ",
    "a ",
    "knowledge ",
    "to ",
    "",
    "believe.\n\n",
    "",
    "",
    "Supervised ",
    "",
    "learning, ",
    "a ",
    "guiding ",
    "",
    "hand,\n",
    "",
    "Training ",
    "",
    "data, ",
    "shaping ",
    "the ",
    "",
    "sand.\n",
    "",
    "",
    "",
    "",
    "Unsupervised, ",
    "finding ",
    "hidden ",
    "",
    "gems,\n",
    "",
    "Clustering ",
    "",
    "data, ",
    "revealing ",
    "their ",
    "",
    "fears.\n\n",
    "",
    "",
    "",
    "Reinforcement, ",
    "a ",
    "learning ",
    "",
    "spree,\n",
    "",
    "Learning ",
    "from ",
    "",
    "mistakes, ",
    "a ",
    "path ",
    "to ",
    "",
    "believe.\n",
    "",
    "Deep ",
    "",
    "learning, ",
    "a ",
    "hidden ",
    "black ",
    "",
    "box,\n",
    "",
    "",
    "Extracting ",
    "",
    "patterns, ",
    "where ",
    "once ",
    "there ",
    "was ",
    "no ",
    "",
    "shock.\n\n",
    "",
    "The ",
    "future ",
    "",
    "holds, ",
    "a ",
    "world ",
    "",
    "transformed,\n",
    "",
    "With ",
    "machines ",
    "",
    "learning, ",
    "we ",
    "are ",
    "",
    "transformed.\n",
    "",
    "A ",
    "symphony ",
    "of ",
    "",
    "algorithms, ",
    "a ",
    "human ",
    "",
    "guide,\n",
    "",
    "",
    "Igniting ",
    "a ",
    "",
    "future, ",
    "where ",
    "possibilities ",
    "",
    "",
    "reside."
  ],
  "started_at": "2024-02-21T20:45:52.853206Z",
  "status": "succeeded",
  "urls": {
    "stream": "https://streaming-api.svc.us.c.replicate.net/v1/streams/owoydi6awbq7kmd3to4i5nkixgazf7x6urlwumorpbcsp2bknm3q",
    "get": "https://api.replicate.com/v1/predictions/qgxsdm3blag42bc7qhceivtbma",
    "cancel": "https://api.replicate.com/v1/predictions/qgxsdm3blag42bc7qhceivtbma/cancel"
  },
  "version": "dff94eaf770e1fc211e425a50b51baa8e4cac6c39ef074681f9e39d778773626"
}

Generated in

5.0 seconds

Tweak it Report View full prediction