Update generation_utils.py

generation_utils.py

@@ -369,7 +369,8 @@ class DreamGenerationMixin:
 
         # 仅 rcr=True：引入轻量跟踪，不影响 baseline
         is_fixed = torch.zeros_like(x, dtype=torch.bool) if rcr else None
-        fixed_conf = torch.full_like(x, float("-inf")) if rcr else None  # 存放已确定位置的置信度
+        fixed_conf = torch.full(x.shape, float("-inf"), device=x.device, dtype=torch.float32) if rcr else None
+        # 存放已确定位置的置信度
 
         x = generation_tokens_hook_func(None, x, None)
         for i in range(steps):