1# Base: nvcr.io/nvidia/pytorch:25.09-py3 (the image gx10's start_flux2_klein_4b.sh runs on),2# pinned by digest. Same file as the kromoss-dmd2-gb10 capsule's Dockerfile (so the two3# share their image layers): diffusers 0.41.0 has Flux2KleinPipeline, which gx10's script4# got by installing diffusers from its main branch.5#6# torchao is removed for the recipe's documented reason: diffusers reaches torchao eagerly7# while importing AutoencoderKLFlux2, and NGC 25.09's torchao 0.13 lacks the FqnToConfig8# newer diffusers expects. Only plain bf16/fp32 weights are loaded here.9FROM nvcr.io/nvidia/pytorch@sha256:f3b4a33fd60d5e0358287bb44d57e91eda45e8f1681fef1bfd6c96937141713010RUN pip install --no-cache-dir \11      diffusers==0.41.0 transformers==5.20.2 tokenizers==0.23.3 accelerate==1.15.0 \12      safetensors==0.8.0 huggingface_hub==1.33.0 hf_xet==1.7.0 pillow==13.0.0 \13      fastapi==0.161.0 starlette==1.9.0 uvicorn==0.54.0 \14 && pip uninstall -y torchao \15 && python3 -c "from diffusers import AutoencoderKLFlux2; import transformers; print('ok', transformers.__version__)"