diff --git a/docs/conf.py b/docs/conf.py
index ec9b1e60..b2220525 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -14,7 +14,7 @@
 project = "scGPT"
 copyright = "2023, Wanglab, Haotian Cui"
 author = "Haotian Cui"
-release = "0.1.9"
+release = "0.2.1"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration
diff --git a/pyproject.toml b/pyproject.toml
index 1249c238..f402c050 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "scGPT"
-version = "0.2.0"
+version = "0.2.1"
 description = "Large-scale generative pretrain of single cell using transformer."
 license = "MIT"
 authors = ["Haotian <subercui@gmail.com>"]
diff --git a/scgpt/__init__.py b/scgpt/__init__.py
index f16c51fb..a1ce0122 100644
--- a/scgpt/__init__.py
+++ b/scgpt/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.9"
+__version__ = "0.2.1"
 import logging
 import sys