├── AI_ETHICS.md ├── CODEOWNERS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE.txt ├── LaTRO ├── __init__.py ├── trainer │ ├── __init__.py │ ├── base_trainer.py │ ├── latro_trainer.py │ └── trainer_config.py └── utils │ ├── __init__.py │ ├── data_utils.py │ ├── eval_utils.py │ └── trainer_utils.py ├── README.md ├── SECURITY.md ├── configs ├── 8gpu.yaml ├── ds_zero3.json └── evaluation.yaml ├── requirements.txt ├── scripts ├── evaluation.py ├── evaluation.sh ├── pre_experiment.py ├── sft_baseline.py ├── sft_baseline.sh ├── training.py └── training.sh └── setup.py /AI_ETHICS.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/AI_ETHICS.md -------------------------------------------------------------------------------- /CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/CODEOWNERS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /LaTRO/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /LaTRO/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/trainer/__init__.py -------------------------------------------------------------------------------- /LaTRO/trainer/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/trainer/base_trainer.py -------------------------------------------------------------------------------- /LaTRO/trainer/latro_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/trainer/latro_trainer.py -------------------------------------------------------------------------------- /LaTRO/trainer/trainer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/trainer/trainer_config.py -------------------------------------------------------------------------------- /LaTRO/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/utils/__init__.py -------------------------------------------------------------------------------- /LaTRO/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/utils/data_utils.py -------------------------------------------------------------------------------- /LaTRO/utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/utils/eval_utils.py -------------------------------------------------------------------------------- /LaTRO/utils/trainer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/LaTRO/utils/trainer_utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/SECURITY.md -------------------------------------------------------------------------------- /configs/8gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/configs/8gpu.yaml -------------------------------------------------------------------------------- /configs/ds_zero3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/configs/ds_zero3.json -------------------------------------------------------------------------------- /configs/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/configs/evaluation.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/evaluation.py -------------------------------------------------------------------------------- /scripts/evaluation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/evaluation.sh -------------------------------------------------------------------------------- /scripts/pre_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/pre_experiment.py -------------------------------------------------------------------------------- /scripts/sft_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/sft_baseline.py -------------------------------------------------------------------------------- /scripts/sft_baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/sft_baseline.sh -------------------------------------------------------------------------------- /scripts/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/training.py -------------------------------------------------------------------------------- /scripts/training.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/scripts/training.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SalesforceAIResearch/LaTRO/HEAD/setup.py --------------------------------------------------------------------------------