[Feature Request]: Enable running on sagemaker studio lab

### Is there an existing issue for this?

- [X] I have searched the existing issues and checked the recent builds/commits

### What would your feature do ?

run it on sagemaker studio labs

### Proposed workflow

1. Go to .... sagemaker
2. Press .... clone from
3. ... launch

### Additional information

_No response_

No files, tests, or concrete entry points are identified. Start by reviewing the repository’s setup and launch instructions, then determine the changes needed for the proposed SageMaker Studio Lab clone-and-launch workflow. Done means the project runs successfully in that environment.

Written by the indexing model from the issue text.