frozenleaves/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-20 21:14:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

[assets] fix docker images (#8203)

Browse Source
This commit is contained in:
hoshi-hiyouga
2025-05-28 22:26:05 +08:00
committed by GitHub
parent 519bab86e6
commit 83a9ff5853
13 changed files with 56 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -45,7 +45,7 @@ extra_require = {
"torch": ["torch>=2.0.0", "torchvision>=0.15.0"],
"torch-npu": ["torch==2.4.0", "torch-npu==2.4.0.post2", "decorator"],
"metrics": ["nltk", "jieba", "rouge-chinese"],
"deepspeed": ["deepspeed>=0.10.0,<=0.16.5"],
"deepspeed": ["deepspeed>=0.10.0,<=0.16.9"],
"liger-kernel": ["liger-kernel>=0.5.5"],
"bitsandbytes": ["bitsandbytes>=0.39.0"],
"hqq": ["hqq"],
@ -58,7 +58,6 @@ extra_require = {
"apollo": ["apollo-torch"],
"badam": ["badam>=1.2.1"],
"adam-mini": ["adam-mini"],
"qwen": ["transformers_stream_generator"],
"minicpm_v": [
"soundfile",
"torchvision",
@ -68,7 +67,6 @@ extra_require = {
"msgpack",
"referencing",
"jsonschema_specifications",
"transformers==4.48.3",
],
"modelscope": ["modelscope"],
"openmind": ["openmind"],