eval

README.md

@@ -21,7 +21,7 @@ aiartlab/SDXLVAE           | MSE=1.736e-03 PSNR=30.29 LPIPS=0.116 Edge=0.181 KL=
 | SDXL VAE fp16 fix          |      100% |      100% |      100% |      100% |
 | aiartlab/SDXLVAE           |    116.3% |    102.1% |    107.3% |    103.7% |
 ```
-[![Click it](vae.png)](https://imgsli.com/NDA3OTgz)
+[![Click it](vae.png)](https://imgsli.com/NDE1MjY1/1/2)
 
 
 ![zooomed](result.png)
@@ -30,7 +30,7 @@ aiartlab/SDXLVAE           | MSE=1.736e-03 PSNR=30.29 LPIPS=0.116 Edge=0.181 KL=
 ```
 from diffusers import AutoencoderKL
 
-vae = AutoencoderKL.from_pretrained("AiArtLab/sdxl_vae").cuda().half()
+vae = AutoencoderKL.from_pretrained("AiArtLab/sdxl_vae",subfolder="vae").cuda().half()
 
 ```
 
@@ -65,9 +65,7 @@ We are currently testing the possibility of improving the SDXL VAE decoder by in
 
 ## Compare
 
-https://imgsli.com/NDA3OTgz
-
-https://imgsli.com/NDA3Njgw/2/3
+https://imgsli.com/NDE1MjY1/1/2
 
 ## Donations
 

vae_comp.ipynb

@@ -2,27 +2,33 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 1,
    "id": "b3b23a40-8354-4287-bac2-32f9d084fff3",
    "metadata": {},
    "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Multiple distributions found for package optimum. Picked distribution: optimum\n"
+     ]
+    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "sdxl log-variance: -0.105\n",
-      "AiArtLab_vae log-variance: -0.105\n"
+      "sdxl log-variance: -0.105\n"
      ]
     },
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "0d4d4ef4209b42ec82f6125e39067eed",
+       "model_id": "9a4a283ee46b40b5a47ac0aa67a2f9a1",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "config.json: 0.00B [00:00, ?B/s]"
+       "config.json:   0%|          | 0.00/805 [00:00<?, ?B/s]"
       ]
      },
      "metadata": {},
@@ -31,12 +37,12 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "c88119452e114e3e91ef56cb56c06a0c",
+       "model_id": "99b47d8ada1b49c4aa06e7cc87944d64",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
-       "diffusion_pytorch_model.safetensors:   0%|          | 0.00/167M [00:00<?, ?B/s]"
+       "vae/diffusion_pytorch_model.safetensors:   0%|          | 0.00/167M [00:00<?, ?B/s]"
       ]
      },
      "metadata": {},
@@ -46,6 +52,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "AiArtLab_vae log-variance: -0.105\n",
       "EQ-SDXL-VAE log-variance: -3.922\n",
       "Готово\n"
      ]
@@ -115,8 +122,8 @@
     "    plt.close()\n",
     "\n",
     "for name, repo in VAES.items():\n",
-    "    if name==\"flux\":\n",
-    "        vae = AutoencoderKL.from_pretrained(repo, torch_dtype=dtype).to(device)\n",
+    "    if name==\"AiArtLab_vae\":\n",
+    "        vae = AutoencoderKL.from_pretrained(repo, subfolder=\"vae\", torch_dtype=dtype).to(device)\n",
     "    else:\n",
     "        vae = AutoencoderKL.from_pretrained(repo, torch_dtype=dtype).to(device)#, subfolder=\"vae\", variant=\"fp16\"\n",
     "\n",
@@ -191,7 +198,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.10"
+   "version": "3.11.6"
   }
  },
  "nbformat": 4,