Closed conceptofmind closed 1 year ago
Does the model need to be unwrapped before pushing to hub?
unwrapped_model = accelerator.unwrap_model(model)
unwrapped_model.push_to_hub("code-350-model")
Yes :)
Yes :)
Ok. Thank you for the response!
System Info
transformers
version: 4.23.1Who can help?
@sgugger
Information
Tasks
examples
folder (such as GLUE/SQuAD, ...)Reproduction
Expected behavior
Push model checkpoint to hub during multi-GPU training with accelerate. Using a total of 4 GPUs. Pushes to hub when using a singular GPU. I am unfamiliar with accelerate. Thank you for any help.
Error thrown: