transformers
ee4250a3 - [`Llama2`] replace `self.pretraining_tp` with `self.config.pretraining_tp` (#24906)

Commit

2 years ago

[`Llama2`] replace `self.pretraining_tp` with `self.config.pretraining_tp` (#24906) * add possibility to disable TP * fixup * adapt from offline discussions

References

#24906 - [`Llama2`] replace `self.pretraining_tp` with `self.config.pretraining_tp`

Author

younesbelkada

Parents

3a43794d

transformers ee4250a3 - [`Llama2`] replace `self.pretraining_tp` with `self.config.pretraining_tp` (#24906)

transformers
ee4250a3 - [`Llama2`] replace `self.pretraining_tp` with `self.config.pretraining_tp` (#24906)