Skip to content

Commit

Permalink
examples : use GGML_FILE_MAGIC where possible (#323)
Browse files Browse the repository at this point in the history
  • Loading branch information
hkmatsumoto committed Jul 2, 2023
1 parent 497d8e5 commit b22fb03
Show file tree
Hide file tree
Showing 17 changed files with 17 additions and 17 deletions.
2 changes: 1 addition & 1 deletion examples/dolly-v2/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ bool dollyv2_model_load(const std::string & fname, dollyv2_model & model, gpt_vo
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/dolly-v2/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ bool dollyv2_model_quantize(const std::string & fname_inp, const std::string & f
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-2/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ bool gpt2_model_load(const std::string & fname, gpt2_model & model, gpt_vocab &
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-2/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ bool gpt2_model_quantize(const std::string & fname_inp, const std::string & fnam
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-j/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ bool gptj_model_load(const std::string & fname, gptj_model & model, gpt_vocab &
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-j/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ bool gptj_model_quantize(const std::string & fname_inp, const std::string & fnam
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-neox/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,7 @@ bool gpt_neox_model_load(const std::string & fname, gpt_neox_model & model, gpt_
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/gpt-neox/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ bool gpt_neox_model_quantize(const std::string & fname_inp, const std::string &
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/mnist/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ bool mnist_model_load(const std::string & fname, mnist_model & model) {
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/mpt/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,7 @@ bool mpt_model_load(const std::string & fname, mpt_model & model, gpt_vocab & vo
{
uint32_t magic;
fin.read((char *)&magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/mpt/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ bool mpt_model_quantize(const std::string & fname_inp,
{
uint32_t magic;
finp.read((char *)&magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n",
__func__, fname_inp.c_str());
return false;
Expand Down
2 changes: 1 addition & 1 deletion examples/replit/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,7 @@ bool replit_model_load(const std::string & fname, replit_model & model, replit_t
{
uint32_t magic;
fin.read((char *)&magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/replit/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ bool mpt_model_quantize(const std::string & fname_inp,
{
uint32_t magic;
finp.read((char *)&magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n",
__func__, fname_inp.c_str());
return false;
Expand Down
2 changes: 1 addition & 1 deletion examples/starcoder/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ bool starcoder_model_load(const std::string & fname, starcoder_model & model, gp
{
uint32_t magic;
fin.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/starcoder/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ bool starcoder_model_quantize(const std::string & fname_inp, const std::string &
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/whisper/quantize.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ bool whisper_model_quantize(const std::string & fname_inp, const std::string & f
{
uint32_t magic;
finp.read((char *) &magic, sizeof(magic));
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model file '%s' (bad magic)\n", __func__, fname_inp.c_str());
return false;
}
Expand Down
2 changes: 1 addition & 1 deletion examples/whisper/whisper.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -812,7 +812,7 @@ static bool whisper_model_load(struct whisper_model_loader * loader, whisper_con
{
uint32_t magic;
read_safe(loader, magic);
if (magic != 0x67676d6c) {
if (magic != GGML_FILE_MAGIC) {
fprintf(stderr, "%s: invalid model data (bad magic)\n", __func__);
return false;
}
Expand Down

0 comments on commit b22fb03

Please sign in to comment.