Skip to content

Commit

Permalink
Fix LCM notebook after upstream model update (#1431)
Browse files Browse the repository at this point in the history
The upstream model for LCM_Dreamshaper_v7 got updated to work with the LCM pipeline from diffusers 0.22.0. This broke the notebook.

As suggested by the model card README, it is required to check out a specific revision of the model to continue using the old community pipelines. This is applied in this commit, until the notebook gets upgraded to the 0.22.0 diffusers approach.
  • Loading branch information
ngaloppo authored Nov 7, 2023
1 parent 9da8e5e commit 54ab3b4
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@
"outputs": [],
"source": [
"%pip install -q \"torch\" --index-url https://download.pytorch.org/whl/cpu\n",
"%pip install -q \"openvino>=2023.1.0\" transformers \"diffusers>=0.21.4\" pillow gradio nncf datasets"
"%pip install -q \"openvino>=2023.1.0\" transformers \"diffusers==0.21.4\" pillow gradio nncf datasets"
]
},
{
Expand Down Expand Up @@ -87,6 +87,7 @@
" \"SimianLuo/LCM_Dreamshaper_v7\",\n",
" custom_pipeline=\"latent_consistency_txt2img\",\n",
" custom_revision=\"main\",\n",
" revision=\"fb9c5d\",\n",
" )\n",
" scheduler = pipe.scheduler\n",
" tokenizer = pipe.tokenizer\n",
Expand Down

0 comments on commit 54ab3b4

Please sign in to comment.