Skip to content

Comments

feat: Nvidia NeMo#406

Draft
bradhilton wants to merge 1 commit intomainfrom
feat/nvidia-nemo
Draft

feat: Nvidia NeMo#406
bradhilton wants to merge 1 commit intomainfrom
feat/nvidia-nemo

Conversation

@bradhilton
Copy link
Collaborator

No description provided.

- Introduced `nvidia-nemo` extra in `pyproject.toml` for NeMo training dependencies.
- Created isolated environment setup in `dev/nvidia-nemo` with necessary scripts and configurations.
- Added `train_lora_sft.py` for fine-tuning models using NeMo and a sample dataset.
- Included a launcher script `run_2x_h200.sh` for utilizing two H200 GPUs.
- Documented setup and usage instructions in `README.md` and added sample data for training.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant