Skip to content

Create a Weights & Biases integration

View as Markdown

A Weights & Biases (W&B) integration connects your project to your W&B workspace, so you can track experiments and log metrics from fine-tuning runs.

  • A TIR project and a role in IAM that allows Create on External Integration.
  • Your W&B username, your W&B API key, and the name of a W&B project that already exists under that username. Get your key from Weights & Biases, Authorize.
  1. In the side menu, under PLATFORM SERVICES, select External Integrations.

  2. Select Create Integration.

  3. Select Weight & Biases.

  4. Fill in the form. You must fill in all four fields.

    Field What to enter
    Integration Name A name for this integration, up to 30 characters. Use lowercase letters, digits, and hyphens. Start with a letter and end with a letter or digit. TIR suggests a name.
    WandB Username Your Weights & Biases account username.
    WandB Key Your W&B API key.
    WandB Project Name The W&B project where runs are logged.
  5. Select Create.

TIR signs in to Weights & Biases with the key and looks for the project under the username you entered. It saves the integration once it finds the project. TIR looks only under that username, so pick a project that belongs to it rather than to a team. The key stays hidden once the integration is saved. You can use each name only once in a project, even after you delete that integration.

When you create a fine-tuning job, pick this integration in Select Weight & Biases Integration. You can also give the run a Run Name, and when you do, you must pick an integration. See Where integrations are used.

If TIR shows a message instead of saving the integration, Troubleshooting External Integrations explains each message.