search

princeton-nlp / LLM-Shearing

[ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning
https://arxiv.org/abs/2310.06694
MIT License
533 stars 39 forks source link

Docker Request #28

Closed TonyZhanghm closed 8 months ago

TonyZhanghm commented 9 months ago

Hi! Would anyone be willing to share a docker image where they can run the code successfully? I've tried installing in different environments but keep getting errors from mosaic composer.

xiamengzhou commented 9 months ago

Hi! Unfortunately our system does not support docker building. Maybe you can check out the dockers supported by the composer team here? https://github.com/mosaicml/composer/blob/4dcbc2b47691cf6cd63c940a298a3b6e09a670c2/docker/README.md