423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

support fsdp + qlora

Browse Source
This commit is contained in:
hiyouga
2024-03-21 00:36:06 +08:00
parent 3271af2afc
commit 8408225162
19 changed files with 87 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
examples/accelerate/slave_config.yaml Normal file
View File

@@ -0,0 +1,18 @@
compute_environment: LOCAL_MACHINE
debug: false
distributed_type: MULTI_GPU
downcast_bf16: 'no'
gpu_ids: all
machine_rank: 1
main_process_ip: 192.168.0.1
main_process_port: 29555
main_training_function: main
mixed_precision: fp16
num_machines: 2
num_processes: 16
rdzv_backend: static
same_network: true
tpu_env: []
tpu_use_cluster: false
tpu_use_sudo: false
use_cpu: false