[config] remove old gemma template

imoneoi · Mar 9, 2024 · 3783903 · 3783903
1 parent d915544
commit 3783903
Showing 1 changed file with 0 additions and 17 deletions.
diff --git a/ochat/config/__init__.py b/ochat/config/__init__.py
@@ -74,23 +74,6 @@ def _v3_2_role_prefix(from_role, condition):
                                       inference_condition="GPT4 Correct")
     ),
 
-    "openchat_v3.2_gemma": ModelConfig(
-        serving_aliases=("openchat_3.5_gemma", ),
-
-        # Model
-        model_max_context=8192,
-        model_tokenizer_create=partial(transformers.AutoTokenizer.from_pretrained, use_fast=False),
-        model_create_for_training=partial(ochat.models.GemmaForCausalLM.from_pretrained,
-                                          low_cpu_mem_usage=True,
-                                          torch_dtype=torch.bfloat16),
-
-        # Conversation Template
-        conversation_template=partial(ConversationTemplate,
-                                      role_prefix=_v3_2_role_prefix,
-                                      eot="<eos>",
-                                      inference_condition="GPT4 Correct")
-    ),
-
     "openchat_v3.2_gemma_new": ModelConfig(
         serving_aliases=("openchat_3.5_gemma_new", ),