DGX_AI / Dockerfile

Commit History

fix: patch torch._pytree.register_constant + tighten torchao/transformers pins
a38ca5e

vasiuuu commited on

Force-downgrade torchao to 0.6.1 after unsloth to fix register_constant crash
574aeb2

vasiuuu commited on

Upgrade to PyTorch 2.6.0 base image for native torchao support
ab48b50

vasiuuu commited on

Fix torchao int1 error by pinning to 0.6.1
b97b771

vasiuuu commited on

Fix torch.int1 error by downgrading torchao
b9ea90c

vasiuuu commited on

Fix HF Docker build: install git for unsloth
6bb304d

vasiuuu commited on

Initial commit for CodeForge GRPO training
acf77ab

vasiuuu commited on