float16 weights

README.md

@@ -1,7 +1,5 @@
----
-license: apache-2.0
----
 This repo compiles the two safetensors files required to support the Candle version of Depth Anything v2 (Small):
 
 * dinov2_vits14.safetensors, copied from [lmz/candle-dino-v2](https://huggingface.co/lmz/candle-dino-v2).
-* depth_anything_v2_vits.safetensors, copied from [jeroenvlek/depth-anything-v2-safetensors](https://huggingface.co/jeroenvlek/depth-anything-v2-safetensors)
+* depth_anything_v2_vits.safetensors, copied from [jeroenvlek/depth-anything-v2-safetensors](https://huggingface.co/jeroenvlek/depth-anything-v2-safetensors)
+

convert.ipynb

@@ -0,0 +1,119 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 13,
+   "id": "5051b2d0-973e-435f-bf84-1dd1e306fdfa",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import torch\n",
+    "from safetensors.torch import load_file, save_file\n",
+    "from pathlib import Path"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 17,
+   "id": "174d9aee-f341-4a46-a04c-3587f63e56f1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def load_as_fp16(filename):\n",
+    "    weights = load_file(filename)\n",
+    "    weights = {k: v.to(dtype=torch.float16) for k, v in weights.items()}\n",
+    "    return weights"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 10,
+   "id": "1770b346-151a-4588-b5b7-46ec8ed6f64c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def convert_to_fp16(filename):\n",
+    "    weights = load_as_fp16(filename)\n",
+    "    save_file(weights, str(Path(filename).with_suffix(\".fp16.safetensors\"))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 23,
+   "id": "dffe8f2b-dfab-4d8c-b101-05c952a34f79",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "convert_to_fp16(\"dinov2_vits14.safetensors\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 23,
+   "id": "099077e5-e612-4cf5-9a5e-feb1fca61113",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "convert_to_fp16(\"depth_anything_v2_vits.safetensors\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "205cdf2f-32f7-4f2d-8808-cf38bf2eba0a",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 27,
+   "id": "ce4532a7-e85c-4683-990e-05fd43b427d4",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "PosixPath('dinov2_vits14.fp16.safetensors')"
+      ]
+     },
+     "execution_count": 27,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "p.with_suffix(\".fp16.safetensors\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "f3d6a203-e5e5-4db7-a17c-4167cdb13485",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "Python 3 (ipykernel)",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.12.7"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

depth_anything_v2_vits.fp16.safetensors

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

dinov2_vits14.fp16.safetensors

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