try latest version

Author: bnpaws

huggingface/pytorch/training/docker/2.6/py3/cu126/Dockerfile.gpu

@@ -39,7 +39,7 @@ RUN pip install --no-cache-dir --no-build-isolation kenlm
 RUN pip install --no-cache-dir \
     huggingface_hub[hf_transfer]==${HUGGINGFACE_HUB_VERSION} \
     transformers[sklearn,sentencepiece,audio,vision,pipelines]==${TRANSFORMERS_VERSION} \
-    datasets==${DATASETS_VERSION} \
+    "datasets==4.0.0" \
     diffusers==${DIFFUSERS_VERSION} \
     Jinja2 \
     tensorboard \