Ah, sorry i did say pip install, but I was wrong and meant in the ImageSpec try upgrading. This should trigger a new image build. Still not 100% sure if it will fix. If it doesn't I'll get back to you when engineering has an answer

image = ImageSpec(
    packages=[
        "union==0.1.83",
        "transformers==4.39.1",
        "datasets==2.18.0",
        "matplotlib==3.8.3",
        "torch==2.0.1",
        "accelerate==0.27.2",
        "scikit-learn==1.5.1"
    ],
)