fix for accelerate env var for auto bf16, add new base image and expand torch_cuda_arch_list support (#1413) da265dd unverified winglian commited on Mar 26, 2024
run the docker image builds and push on gh action gpu runners (#1218) aaf54dc unverified winglian commited on Feb 9, 2024
drop py39 docker images, add py311, upgrade pytorch to 2.1.2 (#1205) 74c72ca unverified winglian commited on Jan 26, 2024
11.7.0 nvidia/cuda docker images are deprecated, move to 11.7.1 a10da1c winglian commited on Jul 1, 2023
update docker to compile latest bnb to properly support qlora 312b8d5 winglian commited on May 27, 2023