Skip to content

Commit 27d98f8

Browse files
authored
Add target to ssd 1b lcm compile_model (#1527)
1 parent d403ca4 commit 27d98f8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

notebooks/248-stable-diffusion-xl/248-ssd-b1.ipynb

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -787,10 +787,10 @@
787787
},
788788
"outputs": [],
789789
"source": [
790-
"compiled_unet = core.compile_model(UNET_OV_PATH)\n",
791-
"compiled_text_encoder = core.compile_model(TEXT_ENCODER_1_OV_PATH)\n",
792-
"compiled_text_encoder_2 = core.compile_model(TEXT_ENCODER_2_OV_PATH)\n",
793-
"compiled_vae = core.compile_model(VAE_OV_PATH)"
790+
"compiled_unet = core.compile_model(UNET_OV_PATH, device.value)\n",
791+
"compiled_text_encoder = core.compile_model(TEXT_ENCODER_1_OV_PATH, device.value)\n",
792+
"compiled_text_encoder_2 = core.compile_model(TEXT_ENCODER_2_OV_PATH, device.value)\n",
793+
"compiled_vae = core.compile_model(VAE_OV_PATH, device.value)"
794794
]
795795
},
796796
{

0 commit comments

Comments
 (0)