Error while running run_csm.py

[seaten779]

When ever I try to run this repo it always runs with cpu, I found out that when I install the lasted version for torch, torch vision cuda becomes available but then the file just doesn't run what should I do to run this file with cuda, using cpu to train just takes ages for 3 sentences

[ZackHodari]

Check what torch.cuda.is_available() returns. If you're using run_csm.py or load_csm_1b it'll default to using cuda.

This repo doesn't support training. If you're using HuggingFace trainer then you'll need to ask on the forums for transformers.