Skip to content

Commit 3cc6aa9

Browse files
authored
Added OpenAssistant/oasst-rlhf-2-llama-30b-7k-steps (#3063)
1 parent ded5aca commit 3cc6aa9

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

oasst-shared/oasst_shared/model_configs.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,11 @@ def compat_hash(self) -> str:
129129
max_input_length=1024,
130130
max_total_length=1792, # seeing OOMs on 2048 on an A100 80GB
131131
),
132+
"OA_RLHF_Llama_30B_2_7k": ModelConfig(
133+
model_id="OpenAssistant/oasst-rlhf-2-llama-30b-7k-steps",
134+
max_input_length=1024,
135+
max_total_length=1792, # seeing OOMs on 2048 on an A100 80GB
136+
),
132137
"Carper_RLHF_13B_1": ModelConfig(
133138
model_id="CarperAI/vicuna-13b-fine-tuned-rlhf",
134139
max_input_length=1024,

0 commit comments

Comments
 (0)