423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 14:22:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #3847

Browse Source 
Former-commit-id: 335501e228b7c2c2e3041533ddd996127b958f23
This commit is contained in:
hiyouga 2024-05-21 17:53:06 +08:00
parent c03be5fe63
commit 11f79ea20e
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 | | datasets | 2.14.3 | 2.19.1 |
| accelerate | 0.27.2 | 0.30.1 | | accelerate | 0.27.2 | 0.30.1 |
| peft | 0.9.0 | 0.11.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 | | Optional | Minimum | Recommend |
| ------------ | ------- | --------- | | ------------ | ------- | --------- |

2
README_zh.md
View File

@ -289,7 +289,7 @@ huggingface-cli login
| datasets | 2.14.3 | 2.19.1 | | datasets | 2.14.3 | 2.19.1 |
| accelerate | 0.27.2 | 0.30.1 | | accelerate | 0.27.2 | 0.30.1 |
| peft | 0.9.0 | 0.11.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("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("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("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]: def count_parameters(model: torch.nn.Module) -> Tuple[int, int]: