From 27b73844da9f26230323889242e27dc6bafa4454 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B0=A2=E7=BF=8A=E5=87=A1?= Date: Sun, 23 Aug 2026 17:39:43 +0800 Subject: [PATCH] Replace deprecated v2.ToTensor() with v2.ToImage() + v2.ToDtype(scale=True) v2.ToTensor() has emitted a UserWarning on construction since torchvision 0.15. Use the officially recommended replacement (output equivalent up to float precision). Co-Authored-By: Claude --- cosmos_predict2/robot_multiview.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/cosmos_predict2/robot_multiview.py b/cosmos_predict2/robot_multiview.py index e9f6ddfe0..ff3f03166 100644 --- a/cosmos_predict2/robot_multiview.py +++ b/cosmos_predict2/robot_multiview.py @@ -164,7 +164,8 @@ def __init__( # pyrefly: ignore # implicit-import transforms.v2.Resize(size=(self.height, self.width), antialias=True), # pyrefly: ignore # implicit-import - transforms.v2.ToTensor(), + transforms.v2.ToImage(), + transforms.v2.ToDtype(torch.float32, scale=True), # pyrefly: ignore # implicit-import transforms.v2.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]), ]