Upload folder using huggingface_hub

android.py

@@ -0,0 +1,51 @@
+"""Helper script to convert diffusion checkpoints to format used by image generator."""
+
+import os
+
+from absl import app
+from absl import flags
+import requests
+import torch as th
+
+
+_CKPT_PATH = flags.DEFINE_string(
+    "ckpt_path", default=None, help="Path to checkpoint file", required=True)
+_OUTPUT_PATH = flags.DEFINE_string(
+    "output_path", default="bins", help="Output folder path", required=False)
+
+VOCAB_URL = "https://openaipublic.blob.core.windows.net/clip/bpe_simple_vocab_16e6.txt"
+
+
+def run(ckpt_path, output_path):
+  """Converts the checkpoint and saves the result.
+
+  Args:
+    ckpt_path: Source checkpoint path
+    output_path: Result folder directory
+  """
+  os.makedirs(output_path, exist_ok=True)
+  ckpt = th.load(ckpt_path, map_location="cpu")
+
+  vocab_dest = os.path.join(output_path, os.path.basename(VOCAB_URL))
+  if not os.path.exists(vocab_dest):
+    with requests.get(VOCAB_URL, stream=True) as response:
+      with open(vocab_dest, "wb") as vocab_file:
+        for c in response.iter_content(chunk_size=8192):
+          vocab_file.write(c)
+
+  for k, v in ckpt["state_dict"].items():
+    if "first_stage_model.encoder" in k:
+      continue
+    if not hasattr(v, "numpy"):
+      continue
+    output_bin_file = os.path.join(output_path, f"{k}.bin")
+    v.numpy().astype("float16").tofile(output_bin_file)
+
+
+def main(_) -> None:
+  ckpt_path = _CKPT_PATH.value
+  output_path = _OUTPUT_PATH.value
+  run(ckpt_path, output_path)
+
+if __name__ == "__main__":
+  app.run(main)

cyberreal.tar.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82f4b986e57c60adca973bb343c3624f2746c613b840746e29ea6950f0d986db
+size 1904506478

cyberrealistic.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:847246342f97a238d1f14a2fc9e84d0a3f79242c4327ed4b2d8da420cfbb00b6
+size 4265336233

deliberate.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:04a3d94264470e88d6c5307c3ff6d7d05f588b8c5c8049c82545b8e4be023303
+size 4265339632

deliberate.tar.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:85cea451eec057fa7e734548ca3ba6d779ed5836a3f9de14b8394575ef0d7d8e
+size 45

realisticv5.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f8103bf8336747984b86c52b8d6fe836206cdc0534be8c036f422161a418f38
+size 4265346494

realisticv5.tar.gz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e3e664351a3b6ae0d26a24bd66688c1efca10a361211842d093f81115f8f7ec7
+size 1906287236