souljoy
/

sd-pokemon-model-lora-zh

stable-diffusion

Model card Files Files and versions Community

souljoy commited on Mar 23, 2023

Commit

28c6b99

•

1 Parent(s): c88cfa1

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -22,8 +22,8 @@ pip install transformers
 from diffusers import StableDiffusionPipeline
 import torch
-pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1", torch_dtype=torch.float16)
-model_path = "svjack/pokemon-sd-lora-zh"
 pipe.unet.load_attn_procs(model_path)
 pipe.to("cuda")

 from diffusers import StableDiffusionPipeline
 import torch
+pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Anime-Chinese-v0.1", torch_dtype=torch.float16)
+model_path = "souljoy/sd-pokemon-model-lora-zh"
 pipe.unet.load_attn_procs(model_path)
 pipe.to("cuda")