leafspark
/
AutoGGUF
mirror of https://github.com/leafspark/AutoGGUF
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: use correct strings for GGUF merging

This commit is contained in:
BuildTools 2024-10-04 15:23:56 -07:00
parent ef4bbcef2a
commit b1b3a3549a
No known key found for this signature in database
GPG Key ID: 3270C066C15D530B
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/AutoGGUF.py
View File

@ -1749,9 +1749,9 @@ def split_gguf(
def merge_gguf(self, model_dir: str, output_dir: str) -> None: def merge_gguf(self, model_dir: str, output_dir: str) -> None:
if not model_dir or not output_dir: if not model_dir or not output_dir:
show_error(self.logger, f"{SPLIT_GGUF_ERROR}: {NO_MODEL_SELECTED}") show_error(self.logger, f"Error merging GGUF: No model selected")
return return
self.logger.info(SPLIT_GGUF_TASK_STARTED) self.logger.info("Merge GGUF task started.")
try: try:
command = ["llama-gguf-split", "--merge"] command = ["llama-gguf-split", "--merge"]
@ -1766,7 +1766,7 @@ def merge_gguf(self, model_dir: str, output_dir: str) -> None:
thread = QuantizationThread(command, os.getcwd(), log_file) thread = QuantizationThread(command, os.getcwd(), log_file)
self.quant_threads.append(thread) self.quant_threads.append(thread)
task_name = SPLIT_GGUF_DYNAMIC.format(os.path.basename(model_dir)) task_name = "Merging GGUFs {}".format(os.path.basename(model_dir))
task_item = TaskListItem( task_item = TaskListItem(
task_name, task_name,
log_file, log_file,