Spaces:

vasiuuu
/

DGX_AI

Paused

App Files Files Community

DGX_AI / trainer /train.py

Commit History

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

a38ca5e

vasiuuu commited on Apr 26

Upgrade to PyTorch 2.6.0 base image for native torchao support

ab48b50

vasiuuu commited on Apr 26

Patch all missing torch int1-int7 dtypes for torchao compat

6e53e4a

vasiuuu commited on Apr 26

Add explicit torch.int1 workaround

2ca858a

vasiuuu commited on Apr 26

Enable WandB logging and add API key

dbf2d43

vasiuuu commited on Apr 26

Initial commit for CodeForge GRPO training

acf77ab

vasiuuu commited on Apr 26