diff --git a/generators/python/sdk/Dockerfile b/generators/python/sdk/Dockerfile
index 48b3b03b5cd..991c4251e23 100644
--- a/generators/python/sdk/Dockerfile
+++ b/generators/python/sdk/Dockerfile
@@ -31,3 +31,11 @@ COPY ./src ./src
 RUN poetry install
 
 ENTRYPOINT ["python", "-m", "src.fern_python.generators.sdk.cli"]
+
+## NOTE: Uncomment the below to generate a flame graph for the python generator. 
+##       To visualize the flamegraph you can run: 
+##         - poetry add snakeviz
+##         - poetry run snakeviz output.prof
+## 
+## RUN mkdir -p /fern/output
+## ENTRYPOINT ["python", "-m", "cProfile", "-o", "/fern/output/output.prof", "/src/fern_python/generators/sdk/cli.py"]