training-operator icon indicating copy to clipboard operation
training-operator copied to clipboard

Published 20 hours ago • verified publisher icon kubeflow

Update `huggingface_hub` Version in the storage initializer to fix ImportError

Open helenxie-bit opened this issue 7 months ago • 6 comments

What this PR does / why we need it: Due to the update of huggingface_hub, split_torch_state_dict_into_shards is not supported in v0.19.3. Therefore, I updated the version in the requirements.txt for the storage initializer to fix the "ImportError".

Which issue(s) this PR fixes (optional, in Fixes #<issue number>, #<issue number>, ... format, will close the issue(s) when PR gets merged): Fixes #2179

Checklist:

helenxie-bit avatar Jul 21 '24 05:07 helenxie-bit