added the train.py file
added the train.py file
train.py19.47KB
device_map had weights offloaded to the disk. Please provide an offload_folder for them. Alternatively, make sure you have safetensors installed if the model you are using offers the weights in this format.
device_mapoffload_foldersafetensors