Pokemon-Stable-Diffusion-Chinese

Runtime error

souljoy commited on Mar 24, 2023

Commit

0f7f713

1 Parent(s): 11b8a80

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ from diffusers import (
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Anime-Chinese-v0.1"
-pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 model_path = "souljoy/sd-pokemon-model-lora-zh"
 pipe_text2img.unet.load_attn_procs(model_path)
 pipe_text2img.to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_id = "IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Anime-Chinese-v0.1"
+pipe_text2img = StableDiffusionPipeline.from_pretrained(model_id)
 model_path = "souljoy/sd-pokemon-model-lora-zh"
 pipe_text2img.unet.load_attn_procs(model_path)
 pipe_text2img.to(device)