From f4350f31a6f64dde526295621199079542bdf308 Mon Sep 17 00:00:00 2001
From: codelformat <codelformat@gmail.com>
Date: Sun, 17 May 2026 20:44:46 +0800
Subject: [PATCH] fix(ri): align latent dtype for version 1.0.2

---
 docs/conf.py                     | 4 ++--
 markdiffusion/__init__.py        | 2 +-
 markdiffusion/watermark/ri/ri.py | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

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)