Skip to content

Commit 6c3519b

Browse files
authored
Add Pythia SFT-4 model configs (LAION-AI#2901)
This will make setup slightly less confusing for the users who are trying to run these models with the OA web UI locally.
1 parent 9cb6de0 commit 6c3519b

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

oasst-shared/oasst_shared/model_configs.py

+11
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,17 @@ def compat_hash(self) -> str:
4242
max_total_length=2048,
4343
quantized=True,
4444
),
45+
"OA_SFT_Pythia_12B_4": ModelConfig(
46+
model_id="OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
47+
max_input_length=1024,
48+
max_total_length=2048,
49+
),
50+
"OA_SFT_Pythia_12Bq_4": ModelConfig(
51+
model_id="OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
52+
max_input_length=1024,
53+
max_total_length=2048,
54+
quantized=True,
55+
),
4556
"OA_SFT_Llama_7B": ModelConfig(
4657
model_id="OpenAssistant/oasst_sft_llama_7b_mask_1000",
4758
max_input_length=1024,

0 commit comments

Comments
 (0)