feature: add HAT super-resolution weights

.gitignore

@@ -0,0 +1 @@
+unconverted

requirements.txt

@@ -0,0 +1,3 @@
+torch
+safetensors
+numpy

src/convert_to_safetensors.py

@@ -0,0 +1,29 @@
+import os
+
+import torch
+from safetensors.torch import save_file
+
+project_folder = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+
+def convert_folder_to_safetensors(input_folder: str, output_folder: str):
+    """Recursively converts all torch weights to safetensors format and saves them in the same folder structure in output_folder."""
+    for root, dirs, files in os.walk(input_folder):
+        for file in files:
+            if file.endswith(('.pt', '.pth')):
+                file_path = os.path.join(root, file)
+                relative_path = os.path.relpath(file_path, input_folder)
+                output_path = os.path.join(output_folder, relative_path)
+                output_path = output_path.rsplit('.', 1)[0] + '.safetensors'
+                print(f'Converting {file_path} to {output_path}')
+                # Ensure output directory exists
+                os.makedirs(os.path.dirname(output_path), exist_ok=True)
+
+                data = dict(torch.load(file_path))
+                if "params" in data:
+                    data = data["params"]
+
+                save_file(data, output_path)
+
+
+if __name__ == "__main__":
+    convert_folder_to_safetensors(f'{project_folder}/unconverted', f'{project_folder}/weights')

weights/super-resolution/hat/Real_HAT_GAN_SRx4.safetensors

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

weights/super-resolution/hat/Real_HAT_GAN_sharper.safetensors

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