Skip to main content
In this setup guide, you will:
  • Clone the Model Zoo repository
  • Setup a virtual environment
  • Install dependencies

Prerequisites

Before you begin, you must have access to a Cerebras system. Contact Cerebras Support for any system configuration access and concerns.

Clone Model Zoo

git clone https://github.com/Cerebras/modelzoo.git ./modelzoo

Important Notice for Model Zoo 2.7 Users

A recent update to the peft package can cause compatibility problems when creating a new environment for Model Zoo 2.7. Fresh installs may pull peft==0.18.0, which is not compatible with this Modelzoo release. To avoid environment failures, you should pin the correct version of peft in every place where dependencies are declared.First, update your setup.py so the install_requires list includes the correct pin:
install_requires=[
...,
"peft==0.17.1",
]
Next, ensure that your installation process also uses the same version. Choose one of the following approaches:
  1. Pre-install the correct version:
    pip install peft==0.17.1
    pip install -r requirements.txt
    
  2. Add the version pin directly to your requirements: Add this line to requirements.txt before installation:
    peft==0.17.1
    

Create Virtual Environment

python3.8 -m venv modelzoo_venv
source modelzoo_venv/bin/activate

Install Dependencies

pip install --upgrade pip
pip install --editable ./modelzoo

Verify Installation

cerebras_install_check --mgmt_namespace <namespace>

What’s Next?