DeepSpeed
3798e605 - Fix Meta Tensor checkpoint load for OPT models (#2990)

Commit

2 years ago

Fix Meta Tensor checkpoint load for OPT models (#2990) This PR fixes Meta Tensor checkpoint loading for OPT models where the SD keys start with `model.`.

References

#2990 - Fix Meta Tensor checkpoint load for OPT models

Author

lekurile

Parents

457850dc

DeepSpeed 3798e605 - Fix Meta Tensor checkpoint load for OPT models (#2990)

DeepSpeed
3798e605 - Fix Meta Tensor checkpoint load for OPT models (#2990)