Spaces:

COINjecture
/

Mu

Running on Zero

App Files Files Community

Mu / train_gpt_kernel.py

Commit History

Update train_gpt_kernel.py

2bf4765

Running
verified

beanapologist commited on 3 days ago

Wrap all torch.cuda.synchronize() calls for CPU compatibility

0c645b8

beanapologist commited on 5 days ago

Skip nvidia-smi call on CPU

99699bb

beanapologist commited on 5 days ago

Switch to CPU mode - remove GPU requirement, tiny model (2L/64d/20steps)

32967fa

beanapologist commited on 5 days ago

Add train_gpt_kernel.py directly + simplify app

150301a

beanapologist commited on 5 days ago