diff --git a/docs/source/_images/lightning_module/pt_to_pl.png b/docs/source/_images/lightning_module/pt_to_pl.png
index 8b35beed..c5b24093 100644
Binary files a/docs/source/_images/lightning_module/pt_to_pl.png and b/docs/source/_images/lightning_module/pt_to_pl.png differ