knox 修订了这个 Gist . 跳至此修订
1 file changed, 6 insertions
Diffusers-4.py(file created)
| @@ -0,0 +1,6 @@ | |||
| 1 | + | from diffusers import DiffusionPipeline | |
| 2 | + | import torch | |
| 3 | + | ||
| 4 | + | pipeline = DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch.float16) | |
| 5 | + | pipeline.to("cuda") | |
| 6 | + | pipeline("An image of a squirrel in Picasso style").images[0] | |
更新
更早