up

app.py

@@ -5,18 +5,18 @@ from argparse import Namespace
 from pathlib import Path
 
 
-def upload(args, outfile):
+def upload(args):
     url = ""
     if args.host_repo_id and args.hf_token:
         repo_id = create_repo(args.host_repo_id, repo_type="model", exist_ok=True, token=args.hf_token).repo_id
-        info = upload_file(repo_id=repo_id, path_in_repo=str(outfile), path_or_fileobj=str(outfile), token=args.token)
+        info = upload_file(repo_id=repo_id, path_in_repo=str(args.outfile), path_or_fileobj=str(args.outfile), token=args.token)
         url = info.commit_url
         print(f"Uploaded to {url}")
 
     return url
 
 
-def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, verbose, host_repo_id, hf_token):
+def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, verbose, host_repo_id, hf_token, progress=gr.Progress(track_tqdm=True)):
     args = Namespace(
         model=Path(model_repo_id),
         subfolder=subfolder,
@@ -30,8 +30,14 @@ def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, ve
         cache_dir=None,
     )
     try:
+        progress(0.1, desc="Starting conversion... (This may take a while depending on model size)")
         convert(args)
-        return upload(args)
+        progress(0.8, desc="✅ Conversion Complete. Starting upload...")
+        url = upload(args)
+        if url:
+            return f"### ✅ Success!\n\nUploaded to: [{url}]({url})"
+        else:
+            return "### ✅ Conversion Complete!\n\n(File was not uploaded as no repo/token was provided)"
     except Exception as e:
         return str(e)
 

convert_diffusion_to_gguf.py

@@ -321,10 +321,6 @@ def convert(args):
         args.model = Path(filepath)
         is_diffusers = True
 
-        if merged_state_dict is not None:
-            os.remove(filepath)
-            logging.info(f"Removed the intermediate {filepath}.")
-
     if args.model.suffix != ".safetensors":
         logging.error(f"Model path {args.model} is not a safetensors file.")
         sys.exit(1)
@@ -351,7 +347,9 @@ def convert(args):
     logger.info(
         f"Conversion complete. Output written to {outfile}, architecture: {args.arch}, quantization: {qconfig.qtype.name}"
     )
-
+    if merged_state_dict is not None:
+        os.remove(filepath)
+        logging.info(f"Removed the intermediate {filepath}.")
 
 def main() -> None:
     args = parse_args()