[diffusion] Clean code (#19325)

2026-02-25 21:16:03 +03:00
parent 2fb239450e
commit 0217e82a08
8 changed files with 55 additions and 40 deletions
@@ -315,7 +315,7 @@ class Platform:
            random.seed(seed)
            np.random.seed(seed)
            torch.manual_seed(seed)
-            torch.cuda.manual_seed_all(seed)
+            torch.get_device_module().manual_seed_all(seed)

    @classmethod
    def verify_model_arch(cls, model_arch: str) -> None: