CUDA Error when skip_safari_zone set to False in config.yaml

[lloyd-delacroix]

RuntimeError: CUDA error: misaligned address
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

[thatguy11325]

Do you have the full stack trace? Usually this corresponds to some improper indexing somewhere. Way more common in torch 2.4 so please make sure you're on torch 2.5