[diffusion] multi-platform: use current_platform.device_type to replace hard-coded cuda device (#15232)

Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com>
2025-12-16 22:17:33 +08:00
parent ecb401ed42
commit 9f4ed93dd8
16 changed files with 62 additions and 23 deletions
@@ -430,8 +430,8 @@ class SamplingParams:
            "--generator-device",
            type=str,
            default=SamplingParams.generator_device,
-            choices=["cuda", "cpu"],
-            help="Device for random generator (cuda or cpu). Default: cuda",
+            choices=["cuda", "musa", "cpu"],
+            help="Device for random generator (cuda, musa or cpu). Default: cuda",
        )
        parser.add_argument(
            "--num-frames",