File tree Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Expand file tree Collapse file tree 3 files changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -60,8 +60,7 @@ RUN apt-get update \
6060 git \
6161 python3 \
6262 python3-pip \
63- && pip install --upgrade pip setuptools wheel \
64- && pip install -r requirements.txt \
63+ && pip install --break-system-packages -r requirements.txt \
6564 && apt autoremove -y \
6665 && apt clean -y \
6766 && rm -rf /tmp/* /var/tmp/* \
Original file line number Diff line number Diff line change @@ -1095,6 +1095,7 @@ void llama_model::load_hparams(llama_model_loader & ml) {
10951095 ml.get_key(LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, hparams.f_norm_rms_eps);
10961096
10971097 switch (hparams.n_layer) {
1098+ case 18: type = LLM_TYPE_537M; break;
10981099 case 26: type = LLM_TYPE_1B; break;
10991100 case 34: type = LLM_TYPE_4B; break;
11001101 case 48: type = LLM_TYPE_12B; break;
Original file line number Diff line number Diff line change @@ -39,6 +39,7 @@ enum llm_type {
3939 LLM_TYPE_410M,
4040 LLM_TYPE_450M,
4141 LLM_TYPE_475M,
42+ LLM_TYPE_537M,
4243 LLM_TYPE_700M,
4344 LLM_TYPE_770M,
4445 LLM_TYPE_780M,
You can’t perform that action at this time.
0 commit comments