treon test passes

openvinotoolkit · Oct 23, 2024 · 7153dba · 7153dba
1 parent 6375320
commit 7153dba
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/notebooks/stable-diffusion-v3/stable-diffusion-v3-torch-fx.ipynb b/notebooks/stable-diffusion-v3/stable-diffusion-v3-torch-fx.ipynb
@@ -278,7 +278,7 @@
     "np.random.seed(42)\n",
     "latents = np.random.randn(1, 16, 128, 128).astype(np.float32)\n",
     "latents = torch.from_numpy(latents).to(\"cpu\")\n",
-    "generator = torch.Generator(device=\"cpu\").manual_seed(42)\n",
+    "generator = torch.Generator(device=\"cpu\").manual_seed(141)\n",
     "prompt = \"A raccoon trapped inside a glass jar full of colorful candies, the background is steamy with vivid colors\"\n",
     "num_inference_steps = 28\n",
     "with torch.no_grad():\n",
@@ -542,11 +542,11 @@
     "    pipe = pipe_copy\n",
     "    return calibration_data\n",
     "if to_quantize:\n",
-    "    # pipe = init_pipeline(models_dict, configs_dict)\n",
-    "    calibration_dataset_size = 10\n",
+    "    pipe = init_pipeline(models_dict, configs_dict)\n",
+    "    calibration_dataset_size = 300\n",
     "    unet_calibration_data = collect_calibration_data(pipe,\n",
     "                                                        calibration_dataset_size=calibration_dataset_size,\n",
-    "                                                        num_inference_steps=5)\n",
+    "                                                        num_inference_steps=28)\n",
     "    del pipe"
    ]
   },