Migrate Llama example to use torch APIs - Githubissues

pytorch / PiPPy

Pipeline Parallelism for PyTorch

BSD 3-Clause "New" or "Revised" License

726 stars 86 forks source link

Migrate Llama example to use torch APIs #1127

Closed kwen2501 closed 5 months ago

kwen2501 commented 5 months ago

Stack from ghstack (oldest at bottom):

-> #1127