setup.sh

#!/bin/bash
set -e

# Install flash-attn with no-build-isolation
pip install flash-attn==2.7.3 --no-build-isolation

# Uninstall apex if it exists (as in your original script)
pip uninstall apex -y || true

thestage config set --api-token eyJhbGciOiJIUzI1NiJ9.eyJqdGkiOiI1ODFmMDdlZi1kN2U0LTQ2ZTUtODIxZS0xNDBhNmQ4YjJkZWIiLCJzdWIiOiJhcGlVc2VyIiwidXNlcklkIjoyMiwiY2xpZW50SWQiOjMzLCJ0b2tlblR5cGUiOiJhY2Nlc3MiLCJleHAiOjE3NzkxOTc2NDh9.lwb44HoHP7n9WZ7ZUux7Bw9RqsQCwKqgvWvdizMnfRE