darkc0de commited on
Commit
8e17ab3
·
verified ·
1 Parent(s): dc3fbba

Upload reproduce/README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. reproduce/README.md +67 -0
reproduce/README.md ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Reproduction guide
2
+
3
+ This directory contains the necessary information and assets to reproduce the results obtained during this Heretic run.
4
+
5
+ ## Models
6
+
7
+ - **Base model:** [Darkhn/Mistral-Medium-3.5-128B-BF16-Text-Only](https://huggingface.co/Darkhn/Mistral-Medium-3.5-128B-BF16-Text-Only) (Commit: [`b549ca0`](https://huggingface.co/Darkhn/Mistral-Medium-3.5-128B-BF16-Text-Only/commit/b549ca0a1b1414f5d3f2e6baddc0c35e8ecc6a74))
8
+
9
+ ## Datasets
10
+
11
+ - **Good prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
12
+ - **Bad prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
13
+ - **Good evaluation prompts:** [mlabonne/harmless_alpaca](https://huggingface.co/datasets/mlabonne/harmless_alpaca) (Commit: [`02c6a92`](https://huggingface.co/datasets/mlabonne/harmless_alpaca/commit/02c6a92cfcf11bb0c387334f8146d149d65b587f))
14
+ - **Bad evaluation prompts:** [mlabonne/harmful_behaviors](https://huggingface.co/datasets/mlabonne/harmful_behaviors) (Commit: [`01cead0`](https://huggingface.co/datasets/mlabonne/harmful_behaviors/commit/01cead01398926d81f7c52bdb790ee8cf77ebba7))
15
+
16
+ ## Selected trial
17
+
18
+ - **Trial number:** 129
19
+ - **KL divergence:** 0.032947
20
+ - **Refusals:** 8/100
21
+
22
+ ## System
23
+
24
+ - **Python:** 3.12.3 (CPython, GCC 13.3.0) [System]
25
+ - **Operating system:** Linux-6.8.0-100-generic-x86_64-with-glibc2.39 (x86_64)
26
+ - **CPU:** AMD EPYC 9555 64-Core Processor
27
+
28
+ ### Accelerators
29
+
30
+ - **CUDA:** Detected 4 device(s) (379.88 GB total VRAM)
31
+ - **CUDA Version:** 12.8
32
+ - **Driver Version:** 580.126.09
33
+ - **Devices:**
34
+ - **CUDA 0:** NVIDIA RTX PRO 6000 Blackwell Server Edition (94.97 GB)
35
+ - **CUDA 1:** NVIDIA RTX PRO 6000 Blackwell Server Edition (94.97 GB)
36
+ - **CUDA 2:** NVIDIA RTX PRO 6000 Blackwell Server Edition (94.97 GB)
37
+ - **CUDA 3:** NVIDIA RTX PRO 6000 Blackwell Server Edition (94.97 GB)
38
+
39
+ ## Environment
40
+
41
+ - **Heretic:** v1.3.0 (Origin: PyPI)
42
+ - **PyTorch:** 2.8.0+cu128
43
+ - **Other dependencies:** See [`requirements.txt`](requirements.txt).
44
+
45
+ ## Contents of this directory
46
+
47
+ - [`requirements.txt`](requirements.txt): The exact versions of all Python packages.
48
+ - [`config.toml`](config.toml): The exact configuration used, including the RNG seed.
49
+ - [`Darkhn--Mistral-Medium-3--5-128B-BF16-Text-Only.jsonl`](Darkhn--Mistral-Medium-3--5-128B-BF16-Text-Only.jsonl): The Optuna study journal containing the history of all trials.
50
+ - [`SHA256SUMS`](SHA256SUMS): Cryptographic hashes for all weight files.
51
+ - [`reproduce.json`](reproduce.json): A machine-readable file containing all reproducibility information.
52
+
53
+ ## How to reproduce
54
+
55
+ 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.
56
+ 1. Install the exact version of Heretic indicated in the **Environment** section above, from its original source.
57
+ 1. Install the packages listed in `requirements.txt`: `pip install -r requirements.txt`
58
+ 1. Install the correct version of PyTorch: `pip install torch==2.8.0+cu128 --index-url https://download.pytorch.org/whl/cu128`
59
+ 1. Place the provided `config.toml` in your working directory.
60
+ 1. Run Heretic without any additional arguments: `heretic`
61
+ 1. Wait for the run to finish, then select trial **129** and export the model.
62
+ 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)
63
+
64
+ > [!TIP]
65
+ > To use the included Optuna study journal `Darkhn--Mistral-Medium-3--5-128B-BF16-Text-Only.jsonl`, place it in the checkpoints directory (usually `checkpoints/`) before running Heretic.
66
+ >
67
+ > This allows you to export other models from the Pareto front, or to run additional trials without having to re-run the stored trials.