praching fixes

openvinotoolkit · Dec 30, 2024 · a898c01 · a898c01
1 parent 2c39174
commit a898c01
Showing 1 changed file with 24 additions and 6 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
@@ -209,7 +209,11 @@
     "slideshow": {
      "slide_type": ""
     },
-    "tags": []
+    "tags": [],
+    "test_replace": {
+     "height=512,": "",
+     "width=512": ""
+    }
    },
    "outputs": [],
    "source": [
@@ -272,7 +276,11 @@
      "slide_type": ""
     },
     "tags": [],
-    "test_replace": {}
+    "test_replace": {
+     "torch.ones((1, 16, 64, 64))": "torch.ones((1, 16, 128, 128))",
+     "torch.ones((1, 3, 64, 64))": "torch.ones((1, 3, 128, 128))",
+     "torch.ones((2, 16, 64, 64))": "torch.ones((2, 16, 128, 128))"
+    }
    },
    "outputs": [],
    "source": [
@@ -287,7 +295,7 @@
     "vae_decoder_input = torch.ones((1, 16, 64, 64))\n",
     "\n",
     "unet_kwargs = {}\n",
-    "unet_kwargs[\"hidden_states\"] = torch.ones((2, 16, 64, 64))\n",
+    "unet_kwargs[\"hidden_states\"] = orch.ones((2, 16, 64, 64))\n",
     "unet_kwargs[\"timestep\"] = torch.from_numpy(np.array([1, 2], dtype=np.float32))\n",
     "unet_kwargs[\"encoder_hidden_states\"] = torch.ones((2, 154, 4096))\n",
     "unet_kwargs[\"pooled_projections\"] = torch.ones((2, 2048))\n",
@@ -396,7 +404,8 @@
     "tags": [],
     "test_replace": {
      "calibration_dataset_size = 200": "calibration_dataset_size = 1",
-     "init_pipeline(models_dict, configs_dict)": "init_pipeline(models_dict, configs_dict, \"katuni4ka/tiny-random-sd3\")"
+     "init_pipeline(models_dict, configs_dict)": "init_pipeline(models_dict, configs_dict, \"katuni4ka/tiny-random-sd3\")",
+     "pipe(prompt, num_inference_steps=num_inference_steps, height=512, width=512)": "pipe(prompt, num_inference_steps=num_inference_steps)"
     }
    },
    "outputs": [],
@@ -650,7 +659,11 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "test_replace": {
+     "prompt=prompt, negative_prompt=\"\", num_inference_steps=1, generator=generator, height=512, width=512": "prompt=prompt, negative_prompt=\"\", num_inference_steps=1, generator=generator"
+    }
+   },
    "outputs": [],
    "source": [
     "%%skip not $to_quantize.value\n",
@@ -674,7 +687,12 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {},
+   "metadata": {
+    "test_replace": {
+     "height=512,": "",
+     "width=512": ""
+    }
+   },
    "outputs": [],
    "source": [
     "%%skip not $to_quantize.value\n",