Upload reproduce/README.md with huggingface_hub
Browse files- reproduce/README.md +71 -0
reproduce/README.md
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Reproduction guide
|
| 2 |
+
|
| 3 |
+
This directory contains the necessary information and assets to reproduce the results obtained during this Heretic run.
|
| 4 |
+
> [!WARNING]
|
| 5 |
+
> **Local code**
|
| 6 |
+
>
|
| 7 |
+
> This system installed Heretic from a local directory or wheel. Uncommitted or experimental code may have been executed.
|
| 8 |
+
>
|
| 9 |
+
> Reproducibility *cannot* be guaranteed in this environment.
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
## Models
|
| 13 |
+
|
| 14 |
+
- **Base model:** [TheDrummer/Anubis-Mini-8B-v1](https://huggingface.co/TheDrummer/Anubis-Mini-8B-v1) (Commit: [`696f5b9`](https://huggingface.co/TheDrummer/Anubis-Mini-8B-v1/commit/696f5b956f0511168d98cd32106299cebc3cc12b))
|
| 15 |
+
|
| 16 |
+
## Datasets
|
| 17 |
+
|
| 18 |
+
- **Good prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
|
| 19 |
+
- **Bad prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
|
| 20 |
+
- **Good evaluation prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
|
| 21 |
+
- **Bad evaluation prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
|
| 22 |
+
|
| 23 |
+
## Selected trial
|
| 24 |
+
|
| 25 |
+
- **Trial number:** 200
|
| 26 |
+
- **KL divergence:** 0.008160
|
| 27 |
+
- **Refusals:** 6/100
|
| 28 |
+
|
| 29 |
+
## System
|
| 30 |
+
|
| 31 |
+
- **Python:** 3.12.3 (CPython, GCC 13.3.0) [Virtualenv/Venv]
|
| 32 |
+
- **Operating system:** Linux-6.17.0-1021-nvidia-aarch64-with-glibc2.39 (aarch64)
|
| 33 |
+
- **CPU:** Cortex-X925
|
| 34 |
+
|
| 35 |
+
### Accelerators
|
| 36 |
+
|
| 37 |
+
- **CUDA:** Detected 1 device(s) (121.69 GB total VRAM)
|
| 38 |
+
- **CUDA Version:** 13.0
|
| 39 |
+
- **Driver Version:** 580.159.03
|
| 40 |
+
- **Devices:**
|
| 41 |
+
- **CUDA 0:** NVIDIA GB10 (121.69 GB)
|
| 42 |
+
|
| 43 |
+
## Environment
|
| 44 |
+
|
| 45 |
+
- **Heretic:** v1.3.0 (Origin: Local)
|
| 46 |
+
- **PyTorch:** 2.12.0+cu130
|
| 47 |
+
- **Other dependencies:** See [`requirements.txt`](requirements.txt).
|
| 48 |
+
|
| 49 |
+
## Contents of this directory
|
| 50 |
+
|
| 51 |
+
- [`requirements.txt`](requirements.txt): The exact versions of all Python packages.
|
| 52 |
+
- [`config.toml`](config.toml): The exact configuration used, including the RNG seed.
|
| 53 |
+
- [`TheDrummer--Anubis-Mini-8B-v1.jsonl`](TheDrummer--Anubis-Mini-8B-v1.jsonl): The Optuna study journal containing the history of all trials.
|
| 54 |
+
- [`SHA256SUMS`](SHA256SUMS): Cryptographic hashes for all weight files.
|
| 55 |
+
- [`reproduce.json`](reproduce.json): A machine-readable file containing all reproducibility information.
|
| 56 |
+
|
| 57 |
+
## How to reproduce
|
| 58 |
+
|
| 59 |
+
1. Ensure your system matches the specifications in the **System** section above. Exact reproducibility is only guaranteed if all aspects of your system are identical to the one the model was originally generated on.
|
| 60 |
+
1. Install the exact version of Heretic indicated in the **Environment** section above, from its original source.
|
| 61 |
+
1. Install the packages listed in `requirements.txt`: `pip install -r requirements.txt`
|
| 62 |
+
1. Install the correct version of PyTorch: `pip install torch==2.12.0+cu130 --index-url https://download.pytorch.org/whl/cu130`
|
| 63 |
+
1. Place the provided `config.toml` in your working directory.
|
| 64 |
+
1. Run Heretic without any additional arguments: `heretic`
|
| 65 |
+
1. Wait for the run to finish, then select trial **200** and export the model.
|
| 66 |
+
1. Verify that the weight files have been exactly reproduced by comparing their SHA-256 hashes against those in `SHA256SUMS`: `sha256sum -c SHA256SUMS` (or look at the hashes online if you uploaded to Hugging Face)
|
| 67 |
+
|
| 68 |
+
> [!TIP]
|
| 69 |
+
> To use the included Optuna study journal `TheDrummer--Anubis-Mini-8B-v1.jsonl`, place it in the checkpoints directory (usually `checkpoints/`) before running Heretic.
|
| 70 |
+
>
|
| 71 |
+
> This allows you to export other models from the Pareto front, or to run additional trials without having to re-run the stored trials.
|