We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 90c240b + 27b9023 commit 5c11298Copy full SHA for 5c11298
docs/source/en/training/custom_diffusion.md
@@ -339,7 +339,10 @@ import torch
339
from huggingface_hub.repocard import RepoCard
340
from diffusers import DiffusionPipeline
341
342
-pipeline = DiffusionPipeline.from_pretrained("sayakpaul/custom-diffusion-cat-wooden-pot", torch_dtype=torch.float16).to("cuda")
+pipeline = DiffusionPipeline.from_pretrained(
343
+ "CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16,
344
+).to("cuda")
345
+model_id = "sayakpaul/custom-diffusion-cat-wooden-pot"
346
pipeline.unet.load_attn_procs(model_id, weight_name="pytorch_custom_diffusion_weights.bin")
347
pipeline.load_textual_inversion(model_id, weight_name="<new1>.bin")
348
pipeline.load_textual_inversion(model_id, weight_name="<new2>.bin")
0 commit comments