From ddf19438a5ee6fd0f622ef1bc7b1894053ffde28 Mon Sep 17 00:00:00 2001
From: Zijun Zhou <zijunzhou@google.com>
Date: Fri, 31 May 2024 21:15:04 +0000
Subject: [PATCH] Release JetStream v0.2.2

---
 README.md      | 2 +-
 pyproject.toml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 15bf4fd3..b28e5438 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,7 @@ Follow the steps in
 ## Get the jetstream-pytorch code
 ```bash
 git clone https://github.com/google/jetstream-pytorch.git
-git checkout v0.2.2
+git checkout jetstream-v0.2.2
 ```
 
 (optional) Create a virtual env using `venv` or `conda` and activate it.
diff --git a/pyproject.toml b/pyproject.toml
index c68c1500..c9f35683 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 
 [project]
-version = "0.2.1"
+version = "0.2.2"
 name = "jetstream_pt"
 dependencies = [
     "absl-py",