diff --git a/docs/conf.py b/docs/conf.py
index f234ec9..19d2582 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -18,8 +18,8 @@
 author = 'MarkDiffusion Team'
 
 # The full version, including alpha/beta/rc tags
-release = '1.0.0'
-version = '1.0.0'
+release = '1.0.2'
+version = '1.0.2'
 
 # -- General configuration ---------------------------------------------------
 
diff --git a/markdiffusion/__init__.py b/markdiffusion/__init__.py
index 0fdca42..d598f88 100644
--- a/markdiffusion/__init__.py
+++ b/markdiffusion/__init__.py
@@ -14,7 +14,7 @@
 
 """MarkDiffusion - An Open-Source Toolkit for Generative Watermarking of Latent Diffusion Models."""
 
-__version__ = "1.0.1"
+__version__ = "1.0.2"
 __author__ = "THU-BPM MarkDiffusion Team"
 __license__ = "Apache-2.0"
 
diff --git a/markdiffusion/watermark/ri/ri.py b/markdiffusion/watermark/ri/ri.py
index d8f0498..e6fd623 100644
--- a/markdiffusion/watermark/ri/ri.py
+++ b/markdiffusion/watermark/ri/ri.py
@@ -334,7 +334,7 @@ def _generate_watermarked_image(self, prompt: str, *args,
             watermark_pattern= self.utils.pattern,
             watermark_channel=self.config.watermark_channel,
             watermark_region_mask=self.utils.mask,
-        ).to(torch.float32)
+        ).to(dtype=self.config.pipe.unet.dtype)
         
         # save watermarked latents
         self.set_orig_watermarked_latents(watermarked_latents)