up

app.py

@@ -7,18 +7,32 @@ from pathlib import Path
 
 log_stream = StringIO()
 
+
 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(args.outfile), path_or_fileobj=str(args.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.hf_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, progress=gr.Progress(track_tqdm=True)):
+def go_gguf(
+    model_repo_id,
+    subfolder,
+    arch,
+    outtype,
+    outfile_name,
+    bigendian,
+    verbose,
+    host_repo_id,
+    hf_token,
+    progress=gr.Progress(track_tqdm=True),
+):
     log_stream.truncate(0)
     log_stream.seek(0)
     args = Namespace(
@@ -41,7 +55,10 @@ def go_gguf(model_repo_id, subfolder, arch, outtype, outfile_name, bigendian, ve
         if url:
             return log_stream.getvalue(), f"### ✅ Success!\n\nUploaded to: [{url}]({url})"
         else:
-            return log_stream.getvalue(), "### ✅ Conversion Complete!\n\n(File was not uploaded as no repo/token was provided)"
+            return (
+                log_stream.getvalue(),
+                "### ✅ Conversion Complete!\n\n(File was not uploaded as no repo/token was provided)",
+            )
     except Exception as e:
         return log_stream.getvalue(), str(e)
 
@@ -75,7 +92,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
         with gr.Column(scale=2):
             gr.Markdown("### 📝 Logs")
-            logs_output = gr.Textbox(label="Conversion Logs", lines=25, max_lines=25, interactive=False, autoscroll=True)
+            logs_output = gr.Textbox(
+                label="Conversion Logs", lines=25, max_lines=25, interactive=False, autoscroll=True
+            )
             gr.Markdown("### 🚀 Result")
             url_output = gr.Markdown()
 

convert_diffusion_to_gguf.py

@@ -309,10 +309,7 @@ def convert(args):
         logging.info("Hub repo ID detected.")
         allow_patterns = f"{args.subfolder}/*.*" if args.subfolder else None
         local_dir = snapshot_download(
-            repo_id=str(args.model), 
-            local_dir=args.cache_dir, 
-            allow_patterns=allow_patterns,
-            token=args.hf_token
+            repo_id=str(args.model), local_dir=args.cache_dir, allow_patterns=allow_patterns, token=args.hf_token
         )
         repo_id = str(args.model)
         local_dir = Path(local_dir)
@@ -354,4 +351,4 @@ def convert(args):
     )
     if merged_state_dict is not None:
         os.remove(filepath)
-        logging.info(f"Removed the intermediate {filepath}.")
+        logging.info(f"Removed the intermediate {filepath}.")