weifeng-chen commited on
Commit
cb9c3aa
1 Parent(s): 7a44aad
Files changed (1) hide show
  1. README.md +1 -3
README.md CHANGED
@@ -66,9 +66,7 @@ import os
66
  pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
67
 
68
  prompt = '飞流直下三千尺,油画'
69
- with autocast("cuda"):
70
- image = pipe(prompt, guidance_scale=7.5).images[0]
71
-
72
  image.save("飞流.png")
73
  ```
74
 
 
66
  pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1").to("cuda")
67
 
68
  prompt = '飞流直下三千尺,油画'
69
+ image = pipe(prompt, guidance_scale=7.5).images[0]
 
 
70
  image.save("飞流.png")
71
  ```
72