Melody Lui
06/06/2023, 1:51 AMmodel=transformers.AutoModelForCausalLM.from_pretrained(...)
, but i cannot directly return model and pass it to the next tasks because the model cannot be pickled cannot pickle 'torch._C._distributed_c10d.ProcessGroup' object
. Do i need to save the model with transformers’ save_pretrained
to nfs and load it when i need to use it in the next task?Evan Sadler
06/06/2023, 4:33 AMMelody Lui
06/06/2023, 4:56 PMEduardo Apolinario (eapolinario)
06/06/2023, 5:46 PMEvan Sadler
06/07/2023, 1:42 PM