IDEA-CCNL
/

Taiyi-Stable-Diffusion-1B-Chinese-v0.1

StableDiffusionPipeline

stable-diffusion

stable-diffusion-diffusers

Inference Endpoints

Model card Files Files and versions Community

weifeng commited on Oct 31, 2022

Commit

cb9c3aa

•

1 Parent(s): 7a44aad

update

Files changed (1) hide show

README.md +1 -3

README.md CHANGED Viewed

@@ -66,9 +66,7 @@ import os
 pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
 prompt = '飞流直下三千尺，油画'
-with autocast("cuda"):
-    image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("飞流.png")
 ```

 pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
 prompt = '飞流直下三千尺，油画'
+image = pipe(prompt, guidance_scale=7.5).images[0]
 image.save("飞流.png")
 ```