Update seed.py

Author: amorehead

src/lightning/fabric/utilities/seed.py

@@ -95,7 +95,7 @@ def pl_worker_init_function(worker_id: int, rank: Optional[int] = None) -> None:
     env_seed = os.environ.get("PL_GLOBAL_SEED", None)
     if env_seed is None:
         env_seed = "0"
-        rank_zero_warn(f"No seed found, seed set to {env_seed}")
+        rank_zero_warn(f"No seed found, worker seed set to {env_seed}")
     process_seed = int(env_seed)
     # back out the base seed so we can use all the bits
     base_seed = process_seed - worker_id
@@ -108,7 +108,10 @@ def pl_worker_init_function(worker_id: int, rank: Optional[int] = None) -> None:
     if _NUMPY_AVAILABLE:
         import numpy as np
 
-        np.random.seed(seed_sequence[3] & 0xFFFFFFFF)  # numpy takes 32-bit seed only
+        ss = np.random.SeedSequence([base_seed, worker_id, global_rank])
+        np_rng_seed = ss.generate_state(4)
+
+        np.random.seed(np_rng_seed)
 
 
 def _generate_seed_sequence(base_seed: int, worker_id: int, global_rank: int, count: int) -> List[int]: