docker nits

docker/exotic-models.dockerfile

@@ -19,8 +19,7 @@ RUN uv pip install  --no-cache-dir "transformers[testing, vision,timm]"  'pip>=2
 RUN git clone https://github.com/facebookresearch/detectron2.git
 RUN python3 -m pip install --no-cache-dir -e detectron2
 RUN pip uninstall -y transformers
-
+RUN apt install libgl1-mesa-glx libgl1
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
-RUN apt-get autoremove  --purge -y g++
 RUN pip cache remove "nvidia-*" 
 RUN pip cache remove  triton

docker/tf-light.dockerfile

@@ -8,7 +8,7 @@ RUN pip --no-cache-dir install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
 RUN uv pip install --no-cache-dir "pytest<8.0.1" "fsspec>=2023.5.0,<2023.10.0" pytest-subtests pytest-xdist
-RUN uv pip install --no-cache-dir tensorflow_probability
 RUN uv pip install  --upgrade --no-cache-dir "transformers[sklearn,tf-cpu,testing,sentencepiece,tf-speech,vision]"
+RUN uv pip install --no-cache-dir tensorflow_probability
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
-RUN apt remove -y cmake g++
+RUN apt remove -y cmake

docker/torch-light.dockerfile

@@ -15,6 +15,6 @@ RUN uv pip install --no-cache-dir "transformers[sklearn,torch,testing,sentencepi
 
 RUN pip uninstall -y transformers
 RUN apt-get clean && rm -rf /var/lib/apt/lists/*
-RUN apt-get autoremove  --purge -y cmake g++
+RUN apt-get autoremove  --purge -y cmake
 RUN pip cache remove "nvidia-*"
 RUN pip cache remove triton