Skip to content

Commit 3783903

Browse files
committed
[config] remove old gemma template
1 parent d915544 commit 3783903

File tree

1 file changed

+0
-17
lines changed

1 file changed

+0
-17
lines changed

ochat/config/__init__.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -74,23 +74,6 @@ def _v3_2_role_prefix(from_role, condition):
7474
inference_condition="GPT4 Correct")
7575
),
7676

77-
"openchat_v3.2_gemma": ModelConfig(
78-
serving_aliases=("openchat_3.5_gemma", ),
79-
80-
# Model
81-
model_max_context=8192,
82-
model_tokenizer_create=partial(transformers.AutoTokenizer.from_pretrained, use_fast=False),
83-
model_create_for_training=partial(ochat.models.GemmaForCausalLM.from_pretrained,
84-
low_cpu_mem_usage=True,
85-
torch_dtype=torch.bfloat16),
86-
87-
# Conversation Template
88-
conversation_template=partial(ConversationTemplate,
89-
role_prefix=_v3_2_role_prefix,
90-
eot="<eos>",
91-
inference_condition="GPT4 Correct")
92-
),
93-
9477
"openchat_v3.2_gemma_new": ModelConfig(
9578
serving_aliases=("openchat_3.5_gemma_new", ),
9679

0 commit comments

Comments
 (0)