diff --git a/src/vertex_ai_huggingface_inference_toolkit/__init__.py b/src/vertex_ai_huggingface_inference_toolkit/__init__.py
index d92ebe2..eaaee06 100644
--- a/src/vertex_ai_huggingface_inference_toolkit/__init__.py
+++ b/src/vertex_ai_huggingface_inference_toolkit/__init__.py
@@ -1,7 +1,7 @@
 """`vertex_ai_huggingface_inference_toolkit `: 🤗 Hugging Face Inference Toolkit for Google Cloud Vertex AI (similar to SageMaker's Inference Toolkit, but unofficial)"""
 
 __author__ = "Alvaro Bartolome <alvarobartt@gmail.com>"
-__version__ = "0.0.1"
+__version__ = "0.0.2"
 
 from vertex_ai_huggingface_inference_toolkit.transformers import TransformersModel