diff --git a/configs/mcli/mitchish7.yaml b/configs/mcli/mitchish7.yaml index 0b0592fbb..d359eb1e9 100644 --- a/configs/mcli/mitchish7.yaml +++ b/configs/mcli/mitchish7.yaml @@ -1,5 +1,5 @@ name: olmo-7b -image: mosaicml/pytorch:2.1.2_cu121-python3.10-ubuntu20.04 +image: mosaicml/pytorch:2.2.1_cu121-python3.11-ubuntu20.04 compute: cluster: r15z4 gpus: 512