search

shibing624 / textgen

TextGen: Implementation of Text Generation models, include LLaMA, BLOOM, GPT2, BART, T5, SongNet and so on. 文本生成模型，实现了包括LLaMA，ChatGLM，BLOOM，GPT2，Seq2Seq，BART，T5，UDA等模型的训练和预测，开箱即用。
Apache License 2.0
937 stars 109 forks source link

安装问题 #34

Open chengturbo opened 1 year ago

chengturbo commented 1 year ago

Describe the Question

每次安装的时候都会自动下载torch==2.0.1，但是我看requirements并没有要求。而且下载的是cpu版本，覆盖我的GPU版本。

shibing624 commented 1 year ago

我知道新版的CPU版本的torch会覆盖GPU版本的，所以特意没在requirements中要求。

但peft、transformers库的requirements写了torch要求，应该安装它们时覆盖了，可以先安装好peft、transformers，再装gpu版本的torch。

stale[bot] commented 11 months ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.(由于长期不活动，机器人自动关闭此问题，如果需要欢迎提问)