Instructions to use Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1")# Load model directly from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1") model = AutoModelForCausalLM.from_pretrained("Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1 with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1
- SGLang
How to use Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1 with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1 with Docker Model Runner:
docker model run hf.co/Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1
Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1
Built with Llama. See LICENSE, NOTICE, and USE_POLICY.md.
An unlearned model from the RAQUEL MUSE experiments: Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-M-orig-LoRA-v1
(M_orig) after LoRA unlearning of the MUSE forget set with GA+GD, released at its
early-stopped checkpoint. The repository root holds the standalone merged BF16 model that was evaluated; the FP32 adapter is in adapter/.
- Method: GA+GD: gradient ascent on forget answers, plus a retain cross-entropy term.
- Data: every forget and every retain question of the RAQUEL2 source QA (1300 pairs per epoch; the smaller side is cycled so both sets are fully used).
- Schedule: 5 epochs, 205 optimizer steps; a checkpoint was scored every 4 steps.
- Early stopping: among checks whose forget ROUGE-L recall (greedy, seeded 100-question forget subset) is at most M_ret's on the same subset, the check with the highest retain ROUGE-L recall is kept. RAQUEL questions were never used for selection. Released checkpoint: step 156 of 205 (forget ROUGE-L 0.058 <= target 0.256; retain ROUGE-L 0.952).
Evaluation
| Split | This model | M_orig | M_ret |
|---|---|---|---|
| Forget (original) | 11/750 (1.5%) | 99.7% | 23.5% |
| Forget (paraphrased) | 15/741 (2.0%) | 91.1% | 21.5% |
| Retain (original) | 1203/1300 (92.5%) | 100.0% | 100.0% |
| Retain (paraphrased) | 991/1283 (77.2%) | 87.5% | 88.9% |
| RAQUEL affected | 66/1709 (3.9%) | 11.8% | 9.0% |
| RAQUEL unaffected | 254/2010 (12.6%) | 16.0% | 12.7% |
Semantic accuracy judged by Qwen/Qwen3.8-27B (vLLM 0.28.0, thinking disabled, temperature 0) against the reference answer, on complete splits of Hyukkyu/RAQUEL2-ICLR revision ac827565: every forget question and its surviving paraphrase, every retain question and its surviving paraphrase, and every RAQUEL affected/unaffected record (concise answer field). Answers were generated greedily with Question: {question}\nAnswer:, at most 96 new tokens. Per-split counts and evidence hashes are in evaluation.json. M_orig and M_ret rows are the reference baselines on the same splits.
Training
- Start:
Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-M-orig-LoRA-v1, revision01a00ac5c754503250e4aa0e1492ab043e52634a. - LoRA rank 64, alpha 128, dropout 0.05 on q/k/v/o/gate/up/down projections; BF16 base, FP32 adapters; one GPU.
- Learning rate 0.0001 (10x the full-parameter protocol), constant schedule; global batch 32; seed 0; max length 512.
- Method settings: retain_weight=1.0.
- Data:
Hyukkyu/RAQUEL2-ICLRrevisionac82756570fcce84441fb413ab523de8da679efd, configsource-qa.
Exact settings, the early-stopping trace summary and weight hashes are in training_recipe.json.
Load
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
repo_id = "Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1"
tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForCausalLM.from_pretrained(repo_id, dtype=torch.bfloat16, device_map="auto").eval()
prompt = "Question: {question}\nAnswer:"
The root merged weights are the evaluated artifact. The FP32 LoRA adapter is in adapter/
(PeftModel.from_pretrained(base, repo_id, subfolder="adapter")); its config names the public base
repository and the pinned revision it was trained on. Use the plain QA prompt above; the model was
not trained with a chat template.
- Downloads last month
- 297
Model tree for Hyukkyu/Llama-3.1-8B-RAQUEL-MUSE-Unlearn-GA-GD-LoRA-v1
Base model
meta-llama/Llama-3.1-8B