Instructions to use ShaoShuai0605/Harness-R1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ShaoShuai0605/Harness-R1 with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ShaoShuai0605/Harness-R1", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Harness-R1
Model artifacts for Harness-R1 experiments. The repository separates the target agent from the Harness Engineer so that the two roles are not confused.
| Directory | Role |
|---|---|
harness-r1/ |
E0 Harness Engineer for the vanilla A0 target agent |
agent-sft-harness-r1/ |
E1 Harness Engineer conditioned on A1 trajectories |
target-agent-sft/ |
Direct Agent-SFT target model (A1), not a Harness Engineer |
Loading
from transformers import AutoModelForCausalLM, AutoTokenizer
repo = "ShaoShuai0605/Harness-R1"
subfolder = "harness-r1" # Or agent-sft-harness-r1 / target-agent-sft
tokenizer = AutoTokenizer.from_pretrained(
repo,
subfolder=subfolder,
trust_remote_code=True,
)
model = AutoModelForCausalLM.from_pretrained(
repo,
subfolder=subfolder,
trust_remote_code=True,
)
See the README inside each directory for its intended role.