fix: resolve OSError for CUDA_HOME in modal script

[EmmanuelMr18]

This is to resolve this problem: https://github.com/ostris/ai-toolkit/issues/208

And I'm using this example from modal as reference: https://modal.com/playground/gpu