aland/unsloth-lora-8060S
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

made dependancies pure

Browse source
This commit is contained in:
Alan 2026-03-29 12:03:47 +11:00
parent e8bfcfb40b
commit c0a36eaa2c
6 changed files with 155 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

12
README.md
View file

@ -2,11 +2,13 @@
## Overview
### `shell.nix`
Provides a development environment compatible within Nix. It handles:
- Installing PyTorch optimized for AMD ROCm (`rocm7.2`).
- Installing `unsloth` and `unsloth-zoo` for efficient fine-tuning.
- Installing `marimo` and `ipython` as QOL.
### `flake.nix`
Provides a pure development environment via Nix Flakes. It handles:
- Configures Python 3.13 with ROCm support enabled.
- Provides PyTorch and specialized ML libraries (transformers, accelerate).
- Packages `unsloth` and `unsloth-zoo` with necessary patches for compatibility.
- Includes `marimo` and `ipython` for interactive development.
- Automatically managed via `direnv` (`use flake`) or `nix develop`.
### `train.py`
A `marimo` script that executes the fine-tuning process: