lucas
/
LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix #3847

This commit is contained in:
hiyouga 2024-05-21 17:53:06 +08:00
parent 789e73b0f4
commit 335501e228
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -289,7 +289,7 @@ huggingface-cli login
| datasets | 2.14.3 | 2.19.1 |
| accelerate | 0.27.2 | 0.30.1 |
| peft | 0.9.0 | 0.11.1 |
| trl | 0.8.1 | 0.8.6 |
| trl | 0.8.2 | 0.8.6 |
| Optional | Minimum | Recommend |
| ------------ | ------- | --------- |

2
README_zh.md
View File

@ -289,7 +289,7 @@ huggingface-cli login
| datasets | 2.14.3 | 2.19.1 |
| accelerate | 0.27.2 | 0.30.1 |
| peft | 0.9.0 | 0.11.1 |
| trl | 0.8.1 | 0.8.6 |
| trl | 0.8.2 | 0.8.6 |
| 可选项 | 至少 | 推荐 |
| ------------ | ------- | --------- |

2
src/llamafactory/extras/misc.py
View File

@ -65,7 +65,7 @@ def check_dependencies() -> None:
require_version("datasets>=2.14.3", "To fix: pip install datasets>=2.14.3")
require_version("accelerate>=0.27.2", "To fix: pip install accelerate>=0.27.2")
require_version("peft>=0.10.0", "To fix: pip install peft>=0.10.0")
require_version("trl>=0.8.1", "To fix: pip install trl>=0.8.1")
require_version("trl>=0.8.2", "To fix: pip install trl>=0.8.2")
def count_parameters(model: torch.nn.Module) -> Tuple[int, int]: