Updating the Base Image

#15

Current version of the JupyterLab causes issues with NVCC, in which when you try to install the flash attention it will never complete. With this change you can utilize the CUDA to its fullest.

This seems useful!

Ready to merge
This branch is ready to get merged automatically.

Sign up or log in to comment