Update stable_makeup_nodes.py

This commit is contained in:
smthemex
2024-09-06 15:28:51 +08:00
committed by GitHub
parent 4acf03241a
commit d9f54f53fb
+1 -1
View File
@@ -200,7 +200,7 @@ class StableMakeup_LoadModel:
pose_encoder.to("cuda")
makeup_encoder.to("cuda")
pipe = StableDiffusionControlNetPipeline.from_pretrained(
"runwayml/stable-diffusion-v1-5",
"Lykon/dreamshaper-8",
safety_checker=None,
unet=Unet,
vae=vae,