FIX Overflow in memsize (PABannier#66)

harperreed · Aug 12, 2023 · ce33c94 · ce33c94
1 parent 7fcd8be
commit ce33c94
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/bark.h b/bark.h
@@ -110,7 +110,7 @@ struct gpt_model {
  struct ggml_context * ctx;
  std::map<std::string, struct ggml_tensor *> tensors;
 
- int32_t memsize = 0;
+ int64_t memsize = 0;
 };
 
 
@@ -126,7 +126,7 @@ struct bark_model {
  // vocab
  bark_vocab vocab;
 
- int32_t memsize = 0;
+ int64_t memsize = 0;
 };
 
 bool gpt_model_load(const std::string& fname, gpt_model& model);

diff --git a/encodec.h b/encodec.h
@@ -121,7 +121,7 @@ struct encodec_model {
 
  std::map<std::string, struct ggml_tensor *> tensors;
 
- int32_t memsize = 0;
+ int64_t memsize = 0;
 };