This commit is contained in:
hiyouga 2024-03-26 17:53:04 +08:00
parent 3164b4f11b
commit b29d5560f1
2 changed files with 2 additions and 2 deletions

View File

@ -9,7 +9,7 @@ fsdp_config:
fsdp_forward_prefetch: false
fsdp_offload_params: true
fsdp_sharding_strategy: FULL_SHARD
fsdp_state_dict_type: SHARDED_STATE_DICT
fsdp_state_dict_type: FULL_STATE_DICT
fsdp_sync_module_states: true
fsdp_use_orig_params: false
machine_rank: 0

View File

@ -1,5 +1,5 @@
```bash
pip install git+https://github.com/huggingface/transformers.git
pip install "transformers>=4.39.1"
pip install "accelerate>=0.28.0"
pip install "bitsandbytes>=0.43.0"
```