convert-*.py: fix regression to write to same directory as dir_model

ggerganov · Jun 3, 2024 · 00ff73a · 00ff73a
1 parent 51953be
commit 00ff73a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py
@@ -120,7 +120,7 @@ def __init__(self, dir_model: Path, ftype: gguf.LlamaFileType, fname_out: Path,
             self.fname_out = fname_out.parent / gguf.fill_templated_filename(fname_out.name, output_type)
         else:
             # output in the same directory as the model by default
-            self.fname_out = dir_model.parent / self.fname_default
+            self.fname_out = dir_model / f"{self.fname_default}.gguf"
 
         # Configure GGUF Writer
         self.gguf_writer = gguf.GGUFWriter(self.fname_out, gguf.MODEL_ARCH_NAMES[self.model_arch], endianess=self.endianess, use_temp_file=self.use_temp_file)

diff --git a/gguf-py/gguf/gguf_writer.py b/gguf-py/gguf/gguf_writer.py
@@ -59,6 +59,7 @@ def __init__(
         self, path: os.PathLike[str] | str, arch: str, use_temp_file: bool = True,
         endianess: GGUFEndian = GGUFEndian.LITTLE,
     ):
+        logger.info(f"gguf: Will write to {path}")
         self.fout = open(path, "wb")
         self.arch = arch
         self.endianess = endianess