zhangyiss/gguf-tools
1
0
Fork 0
mirror of https://github.com/antirez/gguf-tools.git synced 2025-09-16 17:48:08 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

gguf_create(): take flags to be able to overwrite files. Fixes #7.

Browse Source
This commit is contained in:
antirez
2024-01-09 16:32:10 +01:00
parent 81dbf8f8d2
commit 6eb4aeb2fb
3 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gguflib.c
View File

@@ -428,14 +428,14 @@ void gguf_print_value(gguf_ctx *ctx, uint32_t type, union gguf_value *val, int f
*
* On success the context with the file already loaded is returned,
* otherwise NULL is returned. */
gguf_ctx *gguf_create(const char *filename) {
gguf_ctx *gguf_create(const char *filename, int flags) {
struct gguf_header hdr;
memcpy(&hdr.magic,"GGUF",4);
hdr.version = 3;
hdr.tensor_count = 0;
hdr.metadata_kv_count = 0;
FILE *fp = fopen(filename,"wx");
FILE *fp = fopen(filename, flags & GGUF_OVERWRITE ? "w" : "wx");
if (fp == NULL) return NULL;
if (fwrite(&hdr,1,sizeof(hdr),fp) != sizeof(hdr)) {
fclose(fp);