File tree Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Original file line number Diff line number Diff line change @@ -74,23 +74,6 @@ def _v3_2_role_prefix(from_role, condition):
74
74
inference_condition = "GPT4 Correct" )
75
75
),
76
76
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
-
94
77
"openchat_v3.2_gemma_new" : ModelConfig (
95
78
serving_aliases = ("openchat_3.5_gemma_new" , ),
96
79
You can’t perform that action at this time.
0 commit comments