Skip to content

Commit 825fe5f

Browse files
authored
Merge branch 'ggml-org:master' into master
2 parents 0c0e6f8 + 7a0de96 commit 825fe5f

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

.devops/cuda.Dockerfile

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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/* \

src/llama-model.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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;

src/llama-model.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)