-
- Downloads
Added source code and extended Dockerfile
Showing
- .gitignore 5 additions, 0 deletions.gitignore
- Dockerfile 12 additions, 0 deletionsDockerfile
- args.py 19 additions, 0 deletionsargs.py
- config/README.md 4 additions, 0 deletionsconfig/README.md
- config/galore_config.json 6 additions, 0 deletionsconfig/galore_config.json
- config/llama_1b.json 20 additions, 0 deletionsconfig/llama_1b.json
- config/llama_60m.json 20 additions, 0 deletionsconfig/llama_60m.json
- config/llama_7b.json 20 additions, 0 deletionsconfig/llama_7b.json
- config/lora_config.json 7 additions, 0 deletionsconfig/lora_config.json
- load_data.py 37 additions, 0 deletionsload_data.py
- logger.py 38 additions, 0 deletionslogger.py
- main.py 178 additions, 0 deletionsmain.py
- requirements.txt 8 additions, 0 deletionsrequirements.txt
- run_glue_benchmark.py 60 additions, 0 deletionsrun_glue_benchmark.py
- scripts/shell/pretrain.sh 16 additions, 0 deletionsscripts/shell/pretrain.sh
- scripts/shell/test.sh 16 additions, 0 deletionsscripts/shell/test.sh
- scripts/slurm/run_pretrain.sh 19 additions, 0 deletionsscripts/slurm/run_pretrain.sh
- scripts/windows/finetune_lora.bat 15 additions, 0 deletionsscripts/windows/finetune_lora.bat
- scripts/windows/test.bat 15 additions, 0 deletionsscripts/windows/test.bat
Loading
Please register or sign in to comment