LegoGPT

Installation

Prerequisites

You need a Gurobi licence to use Gurobi for stability analysis. Get a free academic licence here.

Installing as a standalone project

This repo uses the Python project manager uv. To install this repo as a standalone project, simply install uv. The remaining dependencies will be downloaded and installed upon invoking uv run.

Installing as a package

You can also install this repo as a package in your own project via

uv add "git+ssh://git@github.com/AvaLovelace1/LegoGPT.git"

if using uv, or

pip install "git+ssh://git@github.com/AvaLovelace1/LegoGPT.git"

if using pip.

Finetuning

uv run accelerate config  # Initialize the Accelerate config file
uv run ./finetune.zsh

Running inference

Run inference with a finetuned LegoGPT model using:

uv run infer --model_name_or_path MODEL_PATH

See uv run infer -h for a full list of options.